ms57rd
/

Res-VMamba

ms57rd commited on Mar 25, 2024

Commit

ac94ddd

verified ·

1 Parent(s): 95bac44

fix class number

Files changed (1) hide show

config.json CHANGED Viewed

@@ -37,7 +37,7 @@ MODEL:
   LABEL_SMOOTHING: 0.1
   MMCKPT: false
   NAME: vssm_small
-  NUM_CLASSES: 241
   PRETRAINED: ./res_vmamba_cnf241_result_2/vssm_small/default/ckpt_epoch_12.pth
   RESUME: ''
   TYPE: vssm

   LABEL_SMOOTHING: 0.1
   MMCKPT: false
   NAME: vssm_small
+  NUM_CLASSES: 1000
   PRETRAINED: ./res_vmamba_cnf241_result_2/vssm_small/default/ckpt_epoch_12.pth
   RESUME: ''
   TYPE: vssm