mlp-project/_ShanghaiA-train.sh
rubberhead 12aabb0d3f More working than not
Not sure if validation works, call it a day
2024-03-03 03:16:54 +00:00

10 lines
No EOL
174 B
Bash

#!/bin/sh
#SBATCH -N 1
#SBATCH -n 1
#SBATCH --partition=Teach-Standard
#SBATCH --gres=gpu:6
#SBATCH --mem=24000
#SBATCH --time=3-00:00:00
python train.py \
--model='stn'