Net2Net icon indicating copy to clipboard operation
Net2Net copied to clipboard

wider function API not consistent

Open savourylie opened this issue 7 years ago • 0 comments

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

savourylie avatar Sep 28 '18 08:09 savourylie