Log In Sign Up
Models:
Marco Lenoir
MarcoTheBlack/
AIplus
0
Downloads: 1
Diff of /HTNet/image-modality/train.sh [000000] .. [1fc74a]

Switch to side-by-side view 

--- a
+++ b/HTNet/image-modality/train.sh
@@ -0,0 +1,11 @@
+#!/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 ./ 
+
+