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

python main.py --input_h 800 --input_w 800 --batch_size 16 --lr 6.25e-5 --gpus 0,1 --load_model /path/of/model_best.pth --data_dir /your/WIDER_train/dir/path