lcsama

Maybe I get the reason why they go opposite ways.

1

Hi, Here is the result when I reproducing your _LeNet.ipynb_ .  I just add a simple `output = F.softmax(output,dim=-1)` before `loss = F.cross_entropy(output, target)` and things changed. We can...