提交 22032aea 编写于 作者: C chenguowei01

update unet.py

上级 c966c2a5
......@@ -64,7 +64,6 @@ class UNet(fluid.dygraph.Layer):
"""
if pretrained_model is not None:
if os.path.exists(pretrained_model):
utils.load_pretrained_model(self.backbone, pretrained_model)
utils.load_pretrained_model(self, pretrained_model)
else:
raise Exception('Pretrained model is not found: {}'.format(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册