forked from QData/C-Tran
-
Notifications
You must be signed in to change notification settings - Fork 2
/
train.sh
30 lines (27 loc) · 1.08 KB
/
train.sh
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
27
28
29
30
#!/bin/bash
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=10
#SBATCH --job-name=train_ctran_bcepoly2
#SBATCH --time=04:00:00
#SBATCH --partition=gpu
#SBATCH --account=kuex0005
#SBATCH --output=train_ctran_bcepoly2.%j.out
#SBATCH --error=train_ctran_bcepoly2.%j.err
module purge
module load gcc/9.3
module load python/3.9.6
module load miniconda/3
module load cuda/11.3
pip install timm
pip install einops
pip install nltk
pip install pillow
pip install numpy
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
pip install -U scikit-learn
pip install pandas
pip install tensorboard
pip install -U albumentations
pip install scikit-multilearn
pip install -U iterative-stratification
python main.py --batch_size 16 --lr 0.00001 --optim 'adam' --layers 3 --dataset 'merged' --use_lmt --grad_ac_step 2 --dataroot /home/kunet.ae/100058256/datasets/ --results_dir /home/kunet.ae/100058256/codes/trained_models/c_tran/ --loss bce_poly --poly_eps 2.0 --img_size 384 --backbone 'densenet' --name 'densenet_bce_poly_2' --run_platform server