fix pretrain download path

2020-04-11 16:35:11 +00:00 · 2020-04-11 16:35:11 +00:00 · 15aaf2af72
parent 353993188a
commit 15aaf2af72
2 changed files with 5 additions and 1 deletions
--- a/tools/download.py
+++ b/tools/download.py
@ -15,7 +15,6 @@
 import sys
 import argparse

-sys.path.append("../")
 from ppcls import model_zoo


--- a/tools/run_download.sh
+++ 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