forked from cabooster/DeepCAD
-
Notifications
You must be signed in to change notification settings - Fork 0
/
o2_flyg.sh
17 lines (15 loc) · 867 Bytes
/
o2_flyg.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#!/bin/bash
#SBATCH --partition gpu_quad # Use a quad GPU
#SBATCH --gres=gpu:rtx8000:1 # Number to use
#SBATCH --time=4-00:00 # Runtime in D-HH:MM format
#SBATCH --mem=100GB # Memory total (for all cores)
#SBATCH -o jobs/deepcad_%j.out # File to which STDOUT will be written, including job ID (%j)
#SBATCH -e jobs/deepcad_%j.err # File to which STDERR will be written, including job ID (%j)
module load gcc/6.2.0
module load cuda/10.2
module load miniconda3/4.10.3
module load python/3.6.0
source activate deepcad
echo "deepcad train initiated"
python script.py train --output_dir '/n/data1/hms/neurobio/wilson/DeepCAD/DeepCAD_results/' --datasets_folder 'deepCAD_data' --datasets_path '/home/ab714/DeepCAD/' --pth_path 'DeepCAD_models'
echo "deepcad train complete"