update csv logging

Author: yuelinxin

csv_logger.py

@@ -1,8 +1,8 @@
 import os
 
-def log_csv(epoch, acc, loss):
+def log_csv(epoch, train_acc, val_acc, loss):
     # check if directory exists
     if not os.path.exists('log'):
         os.makedirs('log')
     with open('log/kt_log.csv', 'a') as f:
-        f.write('{},{},{}\n'.format(epoch, acc, loss))
+        f.write('{},{},{},{}\n'.format(epoch, train_acc, val_acc, loss))

train.py

@@ -128,7 +128,7 @@ def save_checkpoint(state, filename='checkpoint/checkpoint.pth.tar'):
             'optimizer': optimizer.state_dict(),
             'scheduler': scheduler.state_dict()
         })
-    log_csv(epoch, accuracy, epoch_loss)
+    log_csv(epoch, train_accuracy, accuracy, epoch_loss)
 
     scheduler.step()