Merge pull request #2 from Silver-Whiskey/main

VGG16 is working now
google-research · Aug 13, 2021 · 9e6774b · 9e6774b
2 parents 3d5cccb + b0072ab
commit 9e6774b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/third_party/models/vgg.py b/third_party/models/vgg.py
@@ -112,7 +112,7 @@ def vgg16(nchannels=3, **kwargs):
     Args:
         pretrained (bool): If True, returns a model pre-trained on ImageNet
     """
-    model = VGG(make_layers(cfg['D']), nchannels=nchannels, **kwargs)
+    model = VGG(make_layers(cfg['D'], nchannels=nchannels), **kwargs)
     return model