Skip to content
Snippets Groups Projects
Commit ae421b93 authored by l00486551's avatar l00486551
Browse files

fix tinydarknet path in cpu shell

parent 4d568393
Branches
Tags
No related merge requests found
......@@ -15,7 +15,7 @@
# ============================================================================
if [ $# != 1 ] && [ $# != 2 ] && [ $# != 3 ]
then
echo "Usage bash scripts/run_train_cpu.sh [VAL_DATA_DIR] [cifar10|imagenet] [checkpoint_path]"
echo "Usage bash scripts/run_eval_cpu.sh [VAL_DATA_DIR] [cifar10|imagenet] [checkpoint_path]"
exit 1
fi
......
......@@ -37,9 +37,9 @@ fi
BASE_PATH=$(dirname "$(dirname "$(readlink -f $0)")")
if [ $2 == 'imagenet' ]; then
CONFIG_FILE="${BASE_PATH}/imagenet_config.yaml"
CONFIG_FILE="${BASE_PATH}/config/imagenet_config.yaml"
elif [ $2 == 'cifar10' ]; then
CONFIG_FILE="${BASE_PATH}/cifar10_config.yaml"
CONFIG_FILE="${BASE_PATH}/config/cifar10_config.yaml"
else
echo "error: the selected dataset is neither cifar10 nor imagenet"
exit 1
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment