From e51b938bf342cd17fba83e28c1bbd39aa0e202c6 Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Fri, 3 Apr 2020 09:41:34 +0000 Subject: [PATCH] update comment --- models/vgg.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/models/vgg.py b/models/vgg.py index 9102d7d..f46074c 100644 --- a/models/vgg.py +++ b/models/vgg.py @@ -131,9 +131,8 @@ def _vgg(arch, cfg, batch_norm, pretrained, **kwargs): def vgg11(pretrained=False, **kwargs): - r"""VGG 11-layer model from - `"Very Deep Convolutional Networks For Large-Scale Image Recognition" `_ - + """VGG 11-layer model from + Args: pretrained (bool): If True, returns a model pre-trained on ImageNet """ @@ -141,9 +140,8 @@ def vgg11(pretrained=False, **kwargs): def vgg11_bn(pretrained=False, **kwargs): - r"""VGG 11-layer model with batch normalization - `"Very Deep Convolutional Networks For Large-Scale Image Recognition" `_ - + """VGG 11-layer model with batch normalization + Args: pretrained (bool): If True, returns a model pre-trained on ImageNet """ -- GitLab