Skip to content

Commit 44ab09b

Browse files
committed
a bit of readme
1 parent f1ecbef commit 44ab09b

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

README.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ Note:
133133
You can also load the trained model weights and finetune from it, like this:
134134
```
135135
$ export CUDA_VISIBLE_DEVICES=0,1
136-
$ python -m torch.distributed.launch --nproc_per_node=2 tools/train_amp.py --finetune-from ./res/model_final.pth --config ./configs/bisenetv2_city.py # or bisenetv1
136+
$ torchrun --nproc_per_node=2 tools/train_amp.py --finetune-from ./res/model_final.pth --config ./configs/bisenetv2_city.py # or bisenetv1
137137
```
138138

139139

@@ -142,6 +142,10 @@ You can also evaluate a trained model like this:
142142
```
143143
$ python tools/evaluate.py --config configs/bisenetv1_city.py --weight-path /path/to/your/weight.pth
144144
```
145+
or you can use multi gpus:
146+
```
147+
$ torchrun --nproc_per_node=2 tools/evaluate.py --config configs/bisenetv1_city.py --weight-path /path/to/your/weight.pth
148+
```
145149

146150

147151
### Be aware that this is the refactored version of the original codebase. You can go to the `old` directory for original implementation if you need, though I believe you will not need it.

0 commit comments

Comments
 (0)