Code has a few bugs

[ShaofengZou]

Hey, l find that this study is very interesting and helpful.

So I am trying to run your code but find some bugs.

In main.py:

• Line 48,

  from model.vision import LeNet

  Should be:

  from models.vision import LeNet

• Line 51,

  net.apply(weights_init)

  the name 'weights_init' is not defined.

Looking forward to your reply.

[Lyken17]

Thanks for pointing out. Fixed.