Hugging Face's logo

ms57rd
/
Res-VMamba

Transformers
Model card Files
xet
 Community
ms57rd commited on
Commit
ac94ddd
·
verified ·
1 Parent(s): 95bac44

fix class number

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -37,7 +37,7 @@ MODEL:
37
  LABEL_SMOOTHING: 0.1
38
  MMCKPT: false
39
  NAME: vssm_small
40
- NUM_CLASSES: 241
41
  PRETRAINED: ./res_vmamba_cnf241_result_2/vssm_small/default/ckpt_epoch_12.pth
42
  RESUME: ''
43
  TYPE: vssm
 
37
  LABEL_SMOOTHING: 0.1
38
  MMCKPT: false
39
  NAME: vssm_small
40
+ NUM_CLASSES: 1000
41
  PRETRAINED: ./res_vmamba_cnf241_result_2/vssm_small/default/ckpt_epoch_12.pth
42
  RESUME: ''
43
  TYPE: vssm