Net2Net
Net2Net copied to clipboard
wider function API not consistent
Hi Eren!
When trying to run your train_mnist.py script, I ran into the follwoing error:
> Wider Student training ...
Traceback (most recent call last):
File "train_mnist.py", line 159, in <module>
model.net2net_wider()
File "train_mnist.py", line 76, in net2net_wider
self.conv1, self.conv2, _ = wider(self.conv1, self.conv2, 15, noise_var=0.01)
TypeError: wider() got an unexpected keyword argument 'noise_var'
Seems like in the definition you have defined a noise parameter while using noise_var to call it.
Calvin