Merge pull request #4 from wuaodi/patch-1

Update README.md
zifuwan · Apr 17, 2024 · b4dd913 · b4dd913
2 parents ea7c118 + e84cf48
commit b4dd913
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -121,7 +121,7 @@ Install pytorch, cuda and cudnn, then install other dependencies via:
 4. You can also use single-GPU training:
 
     ```shell
-    CUDA_VISIBLE_DEVICES="1,2,3,4,5,6,7" torchrun -m --nproc_per_node=1 train.py -p 29501 -d 0 -n "dataset_name" 
+    CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" torchrun -m --nproc_per_node=1 train.py -p 29501 -d 0 -n "dataset_name" 
     ```
 5. Results will be saved in `log_final` folder.
 

diff --git a/configs/config_pst900.py b/configs/config_pst900.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' # 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'
@@ -104,4 +104,4 @@ def add_path(path):
     args = parser.parse_args()
 
     if args.tensorboard:
-        open_tensorboard()
+        open_tensorboard()