05360171创建于 2022年3月18日历史提交
#!/bin/bash
#SBATCH -p GPU-shared
#SBATCH -t 48:00:00
#SBATCH --gres=gpu:p100:1
#SBATCH --no-requeue

# Usage: ./resume.sh config batch_size resume_file

module load python/3.6.4_gcc5_np1.14.5
module load cuda/9.0

cd $SCRATCH/yolact

python3 train.py --config $1 --batch_size $2 --resume=$3 --save_interval 5000 --start_iter=-1 >>logs/$1_log 2>&1