hey @ptrblck good to have you. you have helped me before, now i am following same code which is  mention above and got this error. can you help me out .
Thanks
Traceback (most recent call last):
File “vggprune.py”, line 127, in 
newmodel = vgg(dataset=args.dataset, cfg=cfg)
File “/home/ustc/akb/network-slimming/models/vgg.py”, line 22, in init
self.feature = self.make_layers(cfg, True)
File “/home/ustc/akb/network-slimming/models/vgg.py”, line 39, in make_layers
conv2d = nn.Conv2d(in_channels, v, kernel_size=3, padding=1, bias=False)
File “/home/ustc/anaconda2/lib/python2.7/site-packages/torch/nn/modules/conv.py”, line 297, in init
False, _pair(0), groups, bias)
File “/home/ustc/anaconda2/lib/python2.7/site-packages/torch/nn/modules/conv.py”, line 38, in init
self.reset_parameters()
File “/home/ustc/anaconda2/lib/python2.7/site-packages/torch/nn/modules/conv.py”, line 44, in reset_parameters
stdv = 1. / math.sqrt(n)
ZeroDivisionError: float division by zero
ustc@ustc-Z10PED16WS:~/akb/network-slimming$