提交 abf5bce4 编写于 作者: L lyuwenyu

update

上级 0102ad56
...@@ -20,17 +20,20 @@ import sys ...@@ -20,17 +20,20 @@ import sys
class _SysPathG(object): class _SysPathG(object):
def __init__(self, path):
self.path = path
def __enter__(self, ): def __enter__(self, ):
sys.path.insert(0, sys.path.insert(0, self.path)
os.path.join(
os.path.dirname(os.path.abspath(__file__)),
'ppcls', 'modeling'))
def __exit__(self, type, value, traceback): def __exit__(self, type, value, traceback):
sys.path.pop(0) _p = sys.path.pop(0)
assert _p == self.path, 'make sure pop {} correctly.'.format(self.path)
with _SysPathG(): with _SysPathG(
os.path.join(
os.path.dirname(os.path.abspath(__file__)), 'ppcls', 'modeling')):
import architectures import architectures
def _load_pretrained_parameters(model, name): def _load_pretrained_parameters(model, name):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册