mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Docs] Fix example_project README (#1575)

pull/1586/head
zzc98 2023-05-22 15:47:03 +08:00 committed by GitHub
parent 1e478462b8
commit b058912c0c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
projects/example_project/README.md
View File

@ -49,19 +49,19 @@ mim train mmpretrain configs/examplenet_8xb32_in1k.py --launcher slurm \
**To test with single GPU:**
```bash
mim test mmpretrain configs/examplenet_8xb32_in1k.py $CHECKPOINT
mim test mmpretrain configs/examplenet_8xb32_in1k.py --checkpoint $CHECKPOINT
```
**To test with multiple GPUs:**
```bash
mim test mmpretrain configs/examplenet_8xb32_in1k.py $CHECKPOINT --launcher pytorch --gpus 8
mim test mmpretrain configs/examplenet_8xb32_in1k.py --checkpoint $CHECKPOINT --launcher pytorch --gpus 8
```
**To test with multiple GPUs by slurm:**
```bash
mim test mmpretrain configs/examplenet_8xb32_in1k.py $CHECKPOINT --launcher slurm \
mim test mmpretrain configs/examplenet_8xb32_in1k.py --checkpoint $CHECKPOINT --launcher slurm \
--gpus 16 --gpus-per-node 8 --partition $PARTITION
```