Commit d33bd65c authored by Tsaris, Aristeidis's avatar Tsaris, Aristeidis
Browse files

format

parent 120a0e83
......@@ -16,18 +16,19 @@ LOG_DIR=summit_logs/
source /gpfs/alpine/stf011/world-shared/atsaris/summit_env/pytorch_1_9_new/activate.sh
jsrun --smpiargs="-disable_gpu_hooks" -n${nnodes} -a1 -c42 -g1 -r1 --bind=proportional-packed:7 --launch_distribution=packed \
bash -c "\
source export_DDP_envvars.sh && \
python -u ../imagenet/main.py \
--arch resnet50 \
-p 10 \
-b 128\
--training-only \
--raport-file ${LOG_DIR}/benchmark.baseline.json \
--epochs 1 \
--prof 100 \
--no-checkpoints \
--data-backend pytorch \
${DATA_DIR}
"
jsrun --smpiargs="-disable_gpu_hooks" -n${nnodes} -a1 -c42 -g1 -r1 \
--bind=proportional-packed:7 --launch_distribution=packed \
bash -c "\
source export_DDP_envvars.sh && \
python -u ../imagenet/main.py \
--arch resnet50 \
-p 10 \
-b 128\
--training-only \
--raport-file ${LOG_DIR}/benchmark.baseline.json \
--epochs 1 \
--prof 100 \
--no-checkpoints \
--data-backend pytorch \
${DATA_DIR}
"
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