diff --git a/tools/eval.py b/tools/eval.py index 7aa370c62bf7ac271f30c955a4b1978a93e51586..768dc4b374a1a871282f3b42b76cf0f30f950dcb 100644 --- a/tools/eval.py +++ b/tools/eval.py @@ -17,6 +17,11 @@ from __future__ import division from __future__ import print_function import os +import sys +__dir__ = os.path.dirname(os.path.abspath(__file__)) +sys.path.append(__dir__) +sys.path.append(os.path.abspath(os.path.join(__dir__, '../'))) + import argparse import paddle.fluid as fluid diff --git a/tools/eval.sh b/tools/eval.sh index 7da3ad4c9411a532966d3544558f87e027050492..a86f56c8f0d4e456560c0495a525e2b36783ceb5 100644 --- a/tools/eval.sh +++ b/tools/eval.sh @@ -1,5 +1,3 @@ -export PYTHONPATH=$PWD:$PYTHONPATH - python -m paddle.distributed.launch \ --selected_gpus="0" \ tools/eval.py \ diff --git a/tools/eval_multi_platform.py b/tools/eval_multi_platform.py index 90f87f7e45aa42feaa3e08047a61903fb4b758cb..4083de6e2fdb442ebd999f704478206844a6a218 100644 --- a/tools/eval_multi_platform.py +++ b/tools/eval_multi_platform.py @@ -17,6 +17,11 @@ from __future__ import division from __future__ import print_function import os +import sys +__dir__ = os.path.dirname(os.path.abspath(__file__)) +sys.path.append(__dir__) +sys.path.append(os.path.abspath(os.path.join(__dir__, '../'))) + import argparse import paddle.fluid as fluid diff --git a/tools/export_model.py b/tools/export_model.py index 763ce473c45d60ba577d65782f5d4737f38a10bc..c04c5018e0965b0e1f5aa5f0d94133a1974ac8db 100644 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -12,6 +12,12 @@ # See the License for the specific language governing permissions and # limitations under the License. +import os +import sys +__dir__ = os.path.dirname(os.path.abspath(__file__)) +sys.path.append(__dir__) +sys.path.append(os.path.abspath(os.path.join(__dir__, '../'))) + import argparse from ppcls.modeling import architectures diff --git a/tools/export_serving_model.py b/tools/export_serving_model.py index e6e4472cdbf8dfd1738dede98b5aa61121f8191a..10632ecd31fe0cf8734edef816d130af2c0738c6 100644 --- a/tools/export_serving_model.py +++ b/tools/export_serving_model.py @@ -14,6 +14,11 @@ import argparse import os +import sys +__dir__ = os.path.dirname(os.path.abspath(__file__)) +sys.path.append(__dir__) +sys.path.append(os.path.abspath(os.path.join(__dir__, '../'))) + from ppcls.modeling import architectures import paddle.fluid as fluid diff --git a/tools/infer/infer.py b/tools/infer/infer.py index 83c1fb4b475d67c116c884319d462558de34caca..ec768fbabf6cefca0d5c82ff7eb80e07b4e2162a 100644 --- a/tools/infer/infer.py +++ b/tools/infer/infer.py @@ -13,13 +13,17 @@ # limitations under the License. import os -import utils +import sys +__dir__ = os.path.dirname(os.path.abspath(__file__)) +sys.path.append(__dir__) +sys.path.append(os.path.abspath(os.path.join(__dir__, '../../'))) + import argparse import numpy as np - import paddle.fluid as fluid from ppcls.modeling import architectures +import utils def parse_args(): diff --git a/tools/run.sh b/tools/run.sh index 5e8043b1205f99dada7448d172f84cd0661df509..ad1e5448904815dc35c689b02f6b7a0094eddd22 100755 --- a/tools/run.sh +++ b/tools/run.sh @@ -1,7 +1,5 @@ #!/usr/bin/env bash -export PYTHONPATH=$PWD:$PYTHONPATH - python -m paddle.distributed.launch \ --selected_gpus="0,1,2,3" \ tools/train.py \ diff --git a/tools/train.py b/tools/train.py index 43c399fa6e0bc5466347e4eaa6d181e8354c3e76..a9c5855799684f6990adfebecabd8df88dc5fa6e 100644 --- a/tools/train.py +++ b/tools/train.py @@ -18,6 +18,11 @@ from __future__ import print_function import argparse import os +import sys +__dir__ = os.path.dirname(os.path.abspath(__file__)) +sys.path.append(__dir__) +sys.path.append(os.path.abspath(os.path.join(__dir__, '../'))) + from sys import version_info import paddle.fluid as fluid diff --git a/tools/train_multi_platform.py b/tools/train_multi_platform.py index 0b1ea2c1beb19ba03b3be39e9a759f45800233c8..4477039301ec547c13716e5f561abf83a3fa7949 100644 --- a/tools/train_multi_platform.py +++ b/tools/train_multi_platform.py @@ -18,6 +18,11 @@ from __future__ import print_function import argparse import os +import sys +__dir__ = os.path.dirname(os.path.abspath(__file__)) +sys.path.append(__dir__) +sys.path.append(os.path.abspath(os.path.join(__dir__, '../'))) + from sys import version_info import paddle.fluid as fluid