update

liouxy · Dec 9, 2017 · 550c6db · 550c6db
1 parent ac519ac
commit 550c6db
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -102,3 +102,4 @@ ENV/
 
 #model
 model
+log/
diff --git a/main.py b/main.py
@@ -131,6 +131,7 @@ def train(training_data_loader, optimizer, model, criterion, epoch, train_log):
         if iteration%10 == 0:
             print("===> Epoch[{}]({}/{}): Loss: {:.10f}".format(epoch, iteration, len(training_data_loader),
                                                                 loss.data[0]))
+            train_log.flush()
 
 
 def test(test_data_loader, model, criterion, epoch, test_log):