As reported in the issue https://github.com/pytorch/examples/issues/215 the example has broken in PyTorch v0.2, especially the LBFGS optimizer behaves strangely, from STEP 4 it only prints one training loss case. Will this be caused by the upgrade of PyTorch?