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