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

Update resnet.py

上级 e6792d34
...@@ -365,7 +365,7 @@ def ResNet34(**args): ...@@ -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. 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. pretrained: bool or str, default: bool=False. Whether to load the pretrained model.
Returns: 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) model = ResNet(config=NET_CONFIG["34"], version="vb", **args)
if isinstance(model.pretrained, bool): if isinstance(model.pretrained, bool):
...@@ -388,7 +388,7 @@ def ResNet34_vd(**args): ...@@ -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. 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. pretrained: bool or str, default: bool=False. Whether to load the pretrained model.
Returns: 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) model = ResNet(config=NET_CONFIG["34"], version="vd", **args)
if isinstance(model.pretrained, bool): if isinstance(model.pretrained, bool):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册