train.sh 90 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
python train.py --data_path "./datasets/flowers" --num_classes 5 --input_size 256 --gpu 0