This argument must be a list

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Dec 11, 2024 · 3c3db34 · 3c3db34
1 parent 7461ffc
commit 3c3db34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/quick_start.md b/docs/quick_start.md
@@ -79,7 +79,7 @@ model = model_factory.build_model(task="segmentation",
             HLSBands.SWIR_1,
             HLSBands.SWIR_2,
         ],
-        necks=[{"name": "SelectIndices", "indices": -1},
+        necks=[{"name": "SelectIndices", "indices": [-1]},
                {"name": "ReshapeTokensToImage"}],
         num_classes=4,
         backbone_pretrained=True,