Skip to content

GitLab

Commit 6f0131ae authored by Thibault de Boissiere's avatar Thibault de Boissiere Committed by Soumith Chintala
Browse files

Update VGG doco (#53) 

* Update VGG doco
parent e56670db
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/models/vgg.py
View file @ 6f0131ae
...@@ -73,7 +73,7 @@ def vgg13_bn(): ...@@ -73,7 +73,7 @@ def vgg13_bn():
def vgg16(): def vgg16():
"""VGG 11-layer model (configuration "B")""" """VGG 16-layer model (configuration "D")"""
return VGG(make_layers(cfg['D'])) return VGG(make_layers(cfg['D']))
...@@ -83,7 +83,7 @@ def vgg16_bn(): ...@@ -83,7 +83,7 @@ def vgg16_bn():
def vgg19(): def vgg19():
"""VGG 19-layer model (configuration "D")""" """VGG 19-layer model (configuration "E")"""
return VGG(make_layers(cfg['E'])) return VGG(make_layers(cfg['E']))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment