File size: 469 Bytes
d617811 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
#!/bin/sh
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 |