fix pretrain download path

15aaf2af · littletomatodonkey · 35399318 · 15aaf2af · 15aaf2af
隐藏空白更改
内联并排

Showing with 5 addition and 1 deletion

tools/download.py tools/download.py +0 -1

tools/run_download.sh tools/run_download.sh +5 -0

未找到文件。
--- 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
+#!/usr/bin/env bash
+
+export PYTHONPATH=$PWD:$PYTHONPATH
+
+python tools/download.py -a ResNet34 -p ./pretrained/ -d 1