File size: 276 Bytes
12f772a |
1 2 3 4 5 6 7 8 9 |
#!/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 70000 \
--validate
|