diff --git a/ppcls/data/reader.py b/ppcls/data/reader.py index 70160a4d65f89a4e97049330f5cb4993dbb35d4f..5749cac13baf8627585b8d0c10bcb86f1ab58144 100755 --- a/ppcls/data/reader.py +++ b/ppcls/data/reader.py @@ -13,16 +13,15 @@ #limitations under the License. import cv2 - import numpy as np import os import signal import paddle -import imaug -from imaug import transform -from imaug import MixupOperator +from . import imaug +from .imaug import transform +from .imaug import MixupOperator from ppcls.utils import logger trainers_num = int(os.environ.get('PADDLE_TRAINERS_NUM', 1)) diff --git a/ppcls/modeling/utils.py b/ppcls/modeling/utils.py index acc4d1948264e72c3cffad52b8139331cfb28420..e7787b8122f086c44e712b46a263a37c91992e00 100644 --- a/ppcls/modeling/utils.py +++ b/ppcls/modeling/utils.py @@ -12,10 +12,11 @@ #See the License for the specific language governing permissions and #limitations under the License. -import architectures import types from difflib import SequenceMatcher +from . import architectures + def get_architectures(): """ diff --git a/ppcls/optimizer/__init__.py b/ppcls/optimizer/__init__.py index 9a192d9a14182185f686dcd68506579926cc8efa..1997992293ab3e7350025c5b43e8619d139fe44f 100644 --- a/ppcls/optimizer/__init__.py +++ b/ppcls/optimizer/__init__.py @@ -15,5 +15,5 @@ from . import optimizer from . import learning_rate -from optimizer import OptimizerBuilder -from learning_rate import LearningRateBuilder +from .optimizer import OptimizerBuilder +from .learning_rate import LearningRateBuilder