update config comment

zifuwan · Apr 17, 2024 · e78f855 · e78f855
1 parent b4dd913
commit e78f855
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/configs/config_MFNet.py b/configs/config_MFNet.py
@@ -46,8 +46,8 @@
 C.norm_std = np.array([0.229, 0.224, 0.225])
 
 """ Settings for network, this would be different for each kind of model"""
-C.backbone = 'sigma_tiny' #'mit_b2' # Remember change the path below.
-C.pretrained_model = None # C.root_dir + '/pretrained/segformer/mit_b2.pth'
+C.backbone = 'sigma_tiny' # sigma_tiny / sigma_small / sigma_base
+C.pretrained_model = None # do not need to change
 C.decoder = 'MambaDecoder' # 'MLPDecoder'
 C.decoder_embed_dim = 512
 C.optimizer = 'AdamW'

diff --git a/configs/config_nyu.py b/configs/config_nyu.py
@@ -87,8 +87,8 @@
 C.norm_std = np.array([0.229, 0.224, 0.225])
 
 """ Settings for network, this would be different for each kind of model"""
-C.backbone = 'sigma_tiny' #'mit_b2' # Remember change the path below.
-C.pretrained_model = None #C.root_dir + '/pretrained/segformer/mit_b2.pth'
+C.backbone = 'sigma_tiny' # sigma_tiny / sigma_small / sigma_base
+C.pretrained_model = None # do not need to change
 C.decoder = 'MambaDecoder' # 'MLPDecoder'
 C.decoder_embed_dim = 512
 C.optimizer = 'AdamW'

diff --git a/configs/config_sunrgbd.py b/configs/config_sunrgbd.py
@@ -84,8 +84,8 @@
 C.norm_std = np.array([0.229, 0.224, 0.225])
 
 """ Settings for network, this would be different for each kind of model"""
-C.backbone = 'sigma_tiny' #'mit_b2' # Remember change the path below.
-C.pretrained_model = None # C.root_dir + '/pretrained/segformer/mit_b2.pth'
+C.backbone = 'sigma_tiny' # sigma_tiny / sigma_small / sigma_base
+C.pretrained_model = None # do not need to change
 C.decoder = 'MambaDecoder' # 'MLPDecoder'
 C.decoder_embed_dim = 512
 C.optimizer = 'AdamW'