blob: a5d16ec250ac55133bd57cf3c19dad9885bce779 [file] [log] [blame]
#!/usr/bin/env bash
# run this experiment with
# nohup bash script/resnet_voc07.sh 0,1 &> resnet_voc07.log &
# to use gpu 0,1 to train, gpu 0 to test and write logs to resnet_voc07.log
gpu=${1:0:1}
export MXNET_CUDNN_AUTOTUNE_DEFAULT=0
export PYTHONUNBUFFERED=1
python train_end2end.py --network resnet --gpu $1
python test.py --network resnet --gpu $gpu