gpus=4 | |
config=$1 | |
output=$2 | |
if [ -z $config ] | |
then | |
echo "No config file found! Run with "sh run.sh [CONFIG_FILE] [OUTPUT_DIR] [OPTS]"" | |
exit 0 | |
fi | |
if [ -z $output ] | |
then | |
echo "No output directory found! Run with "sh run.sh [CONFIG_FILE] [OUTPUT_DIR] [OPTS]"" | |
exit 0 | |
fi | |
shift 2 | |
opts=${@} | |
python train_net.py --config $config \ | |
--num-gpus $gpus \ | |
--dist-url "auto" \ | |
--resume \ | |
OUTPUT_DIR $output \ | |
$opts | |
sh eval.sh $config $output $opts |