File size: 284 Bytes
12f772a |
1 2 3 4 5 6 7 8 9 10 |
#!/bin/bash
work_path=$(dirname $0)
partition=$1
GLOG_vmodule=MemcachedClient=-1 srun --mpi=pmi2 -p $partition -n8 \
--gres=gpu:8 --ntasks-per-node=8 \
python -u main.py \
--config $work_path/config.yaml --launcher slurm \
--load-iter 10000 \
--resume
|