update train & test

AromaticJ · Dec 5, 2020 · 40479d0 · 40479d0
1 parent cad4802
commit 40479d0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -49,14 +49,16 @@ $ python ./CIA-SSD/tools/create_data.py
 Train the CIA-SSD:
 ```bash
 Single GPU
-$ python ./CIA-SSD/tools/train.py
+$ cd ./CIA-SSD/tools
+$ python train.py
 Multiple GPU
-$ python -m torch.distributed.launch --nproc_per_node=4 ./CIA-SSD/tools/train.py
+$ python -m torch.distributed.launch --nproc_per_node=4 train.py
 ```
 
 Evaluate the CIA-SSD:
 ```bash
-$ python ./CIA-SSD/tools/test.py
+$ cd ./CIA-SSD/tools
+$ python test.py
 ```
 
 ## Citation