CTransformer/SBATCH_submit_script.sh at main · chiellini/CTransformer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#!/bin/sh

#SBATCH --mail-type=BEGIN,END,FAIL,ALL
#SBATCH --mail-user=chiellini.lee@gmail.com
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=48
#SBATCH --ntasks-per-node=1
#SBATCH --gpus-per-node=1

#SBATCH --job-name=4_UNet

#python 1_train_segmentation.py --cfg 1_3DUNet_TRAIN
#python 1_train_segmentation.py --cfg 1_DMFNet_TRAIN
#python 1_train_segmentation.py --cfg 1_SwinUNETR_TRAIN
#python 1_train_segmentation.py --cfg 1_VNet_TRAIN

#python 2_5_validate_segmentation.py --cfg 2_DMFNet_VALIDATE --logging_suffix 2_validate
#python 2_5_validate_segmentation.py --cfg 2_VNet_VALIDATE --logging_suffix 2_validate
#python 2_5_validate_segmentation.py --cfg 2_3DUNet_VALIDATE --logging_suffix 2_validate
#python 2_5_validate_segmentation.py --cfg 2_SwinUNETR_VALIDATE --logging_suffix 2_validate
#python 2_5_validate_segmentation.py --cfg 2_DMFNet_VALIDATE --logging_suffix 2_validate


python 4_train_enhancement.py --cfg 4_3DUNet_Enhance_Membrane_Topologically_TRAIN --gpu 0 --logging_suffix 4_topo
#python 4_train_enhancement.py --cfg 4_3DUNet_Enhance_Membrane_TRAIN --gpu 0 --logging_suffix 4_no_topo