docs: 📝 add example script for distributed training

chaofengc · May 30, 2024 · 18f123e · 18f123e
1 parent 036f2cc
commit 18f123e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -219,6 +219,10 @@ python pyiqa/train.py -opt options/train/DBCNN/train_DBCNN.yml
 python pyiqa/train_nsplits.py -opt options/train/DBCNN/train_DBCNN.yml
 ```
 
+Example for distributed training
+```
+torchrun --nproc_per_node=2 --master_port=4321 pyiqa/train.py -opt options/train/CLIPIQA/train_CLIPIQA_koniq10k.yml --launcher pytorch
+```
 
 ## :beers: Contribution