Minor fix for slurm launching scripts of classification (#89)

* support ema for main_deepspeed.py, fix torch.distribute.launch * minor fix

Minor fix for slurm launching scripts of classification (#89)
* support ema for main_deepspeed.py, fix torch.distribute.launch * minor fix
e2728d36 · Zeqiang Lai · zhe chen · adbb02f6 · e2728d36 · e2728d36
Commit e2728d36 authored Apr 11, 2023 by Zeqiang Lai Committed by zhe chen Apr 11, 2023
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

classification/train_in1k.sh classification/train_in1k.sh +1 -1

classification/train_in1k_deepspeed.sh classification/train_in1k_deepspeed.sh +1 -1

No files found.
--- a/classification/train_in1k.sh
+++ b/classification/train_in1k.sh
@@ -23,6 +23,6 @@ PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
    python -u main.py \
    --cfg ${CONFIG} \
    --accumulation-steps 1 \
-    --local_rank 0 \
+    --local-rank 0 \
    --data-path /mnt/lustre/share/images \
    --output work_dirs ${@:4}
--- a/classification/train_in1k_deepspeed.sh
+++ b/classification/train_in1k_deepspeed.sh
@@ -22,6 +22,6 @@ PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
    ${SRUN_ARGS} \
    python -u main_deepspeed.py \
    --cfg ${CONFIG} \
-    --local_rank 0 \
+    --local-rank 0 \
    --data-path /mnt/lustre/share/images \
    --output work_dirs_deepspeed ${@:4}