diff --git a/paddlehub/module/module.py b/paddlehub/module/module.py index 351c8ecbef2cf2ccd4229d404d84a33edf0380e1..aa4392edb9504724e26f67864e56d5fbec78b6d1 100644 --- a/paddlehub/module/module.py +++ b/paddlehub/module/module.py @@ -224,11 +224,14 @@ class Module(object): module_code_version = checker.module_code_version if module_code_version == "v2": - basename = os.path.split(directory)[-1] - dirname = os.path.join(*list(os.path.split(directory)[:-1])) - sys.path.append(dirname) - user_module = importlib.import_module("{}.module".format(basename)) - return user_module.HubModule(directory=directory) + sys.path.insert(0, directory) + # clear module cache + if 'module' in sys.modules: + sys.modules.pop('module') + _module = importlib.import_module("module") + user_module = _module.HubModule(directory=directory) + sys.path.pop(0) + return user_module return ModuleV1(directory=directory) @property