train.sh 112 Bytes
Newer Older
chenych's avatar
Add .sh  
chenych committed
1
2
#!/usr/bin/env bash

3
python main.py --input_h 512 --input_w 512 --batch_size 8 --lr 5e-4 --data_dir ../datasets/