diff --git a/models/vgg.py b/models/vgg.py index 9102d7dbdd44acbb9c6dca2d92c3e3c46b7caaf1..f46074ce8aad2d0e20742d1562b14dcd766572cc 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 """