Update demo_LeNet5_KFAC.py

Author: lixilinx

misc/demo_LeNet5_KFAC.py

@@ -3,7 +3,7 @@
 import torch.nn.functional as F
 import torch.optim as optim
 from torchvision import datasets, transforms
-from kfac import KFAC
+from kfac import KFAC#download from https://gist.github.com/Thrandis/9b3f75a130ec6c24a64117b7d9304c3f
 
 train_loader = torch.utils.data.DataLoader(
         datasets.MNIST('../data', train=True, download=True,           
@@ -69,4 +69,4 @@ def test_loss(model, test_loader):
     lr0 = 0.5*lr0
     optimizer.param_groups[0]['lr'] = lr0
     TestLoss.append(test_loss(model, test_loader))
-    print('Epoch: {}; best test loss: {}'.format(epoch, min(TestLoss)))
+    print('Epoch: {}; best test loss: {}'.format(epoch, min(TestLoss)))