diff --git a/tools/download.py b/tools/download.py index ab88fb44f6118553c6bd3c3816489ae8882ffcb8..157bebccdaa26bbb40a46a5bf065b4e092527af7 100644 --- a/tools/download.py +++ b/tools/download.py @@ -15,7 +15,6 @@ import sys import argparse -sys.path.append("../") from ppcls import model_zoo diff --git a/tools/run_download.sh b/tools/run_download.sh new file mode 100755 index 0000000000000000000000000000000000000000..ba9fa0fc5388db19ed54588ab223ccb0593e33f5 --- /dev/null +++ b/tools/run_download.sh @@ -0,0 +1,5 @@ +#!/usr/bin/env bash + +export PYTHONPATH=$PWD:$PYTHONPATH + +python tools/download.py -a ResNet34 -p ./pretrained/ -d 1