Skip to content

Commit 378990f

Browse files
Merge pull request #271 from IBM/update/docs
updating documentation
2 parents 6b5499b + 43fbfea commit 378990f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

docs/quick_start.md

+1-2
Original file line numberDiff line numberDiff line change
@@ -107,9 +107,8 @@ model_args = dict(
107107
HLSBands.SWIR_1,
108108
HLSBands.SWIR_2,
109109
],
110-
necks=[{"name": "SelectIndices", "indices": -1},
110+
necks=[{"name": "SelectIndices", "indices": [-1]},
111111
{"name": "ReshapeTokensToImage"}],
112-
num_classes=4,
113112
backbone_pretrained=True,
114113
backbone_num_frames=1,
115114
decoder_channels=128,

0 commit comments

Comments
 (0)