Commit e2728d36 authored by Zeqiang Lai's avatar Zeqiang Lai Committed by zhe chen
Browse files

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

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

* minor fix
parent adbb02f6
......@@ -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}
......@@ -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}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment