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

Switch to unified view

a b/HTNet/image-modality/train.sh 

  1
  
#!/bin/bash





  
  

  2
  





  
  

  3
  
export OMP_NUM_THREADS=1





  
  

  4
  





  
  

  5
  
python train.py \





  
  

  6
  
--train-file train.csv \





  
  

  7
  
--val-file valid.csv -b 2 --device cpu \





  
  

  8
  
--lr 0.001 --wd 1.0e-4 --lr-step-size 30 --epochs 80 \





  
  

  9
  
-j 4 --output-dir ./ 





  
  

  10
  





  
  

  11