Update vgg.py (#233)

7492fae4 · Sri Krishna · Soumith Chintala · 25f81c28 · 7492fae4
Commit 7492fae4 authored Aug 26, 2017 by Sri Krishna Committed by Soumith Chintala Aug 26, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

torchvision/models/vgg.py torchvision/models/vgg.py +0 -1

No files found.
--- a/torchvision/models/vgg.py
+++ b/torchvision/models/vgg.py
@@ -54,7 +54,6 @@ class VGG(nn.Module):
                m.weight.data.fill_(1)
                m.bias.data.zero_()
            elif isinstance(m, nn.Linear):
-                n = m.weight.size(1)
                m.weight.data.normal_(0, 0.01)
                m.bias.data.zero_()