未验证 提交 0f3d321b 编写于 作者: C cuicheng01 提交者: GitHub

Update resnet.py

上级 e6792d34
......@@ -365,7 +365,7 @@ def ResNet34(**args):
lr_mult_list: list=[1.0, 1.0, 1.0, 1.0, 1.0]. Control the learning rate of different stages.
pretrained: bool or str, default: bool=False. Whether to load the pretrained model.
Returns:
model: nn.Layer. Specific `ResNet18` model depends on args.
model: nn.Layer. Specific `ResNet34` model depends on args.
"""
model = ResNet(config=NET_CONFIG["34"], version="vb", **args)
if isinstance(model.pretrained, bool):
......@@ -388,7 +388,7 @@ def ResNet34_vd(**args):
lr_mult_list: list=[1.0, 1.0, 1.0, 1.0, 1.0]. Control the learning rate of different stages.
pretrained: bool or str, default: bool=False. Whether to load the pretrained model.
Returns:
model: nn.Layer. Specific `ResNet18_vd` model depends on args.
model: nn.Layer. Specific `ResNet34_vd` model depends on args.
"""
model = ResNet(config=NET_CONFIG["34"], version="vd", **args)
if isinstance(model.pretrained, bool):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册