[1fc74a]: / HTNet / image-modality / train.sh

Download this file

12 lines (7 with data), 202 Bytes

1
2
3
4
5
6
7
8
9
#!/bin/bash
export OMP_NUM_THREADS=1
python train.py \
--train-file train.csv \
--val-file valid.csv -b 2 --device cpu \
--lr 0.001 --wd 1.0e-4 --lr-step-size 30 --epochs 80 \
-j 4 --output-dir ./