minor bug fix

Author: Fangchang Ma

main.py

@@ -4,6 +4,7 @@
 import time
 import sys
 import csv
+import numpy as np
 
 import torch
 import torch.nn as nn
@@ -210,7 +211,7 @@ def main():
         adjust_learning_rate(optimizer, epoch)
 
         # train for one epoch
-        train(train_loader, model, criterion, optimizer, epoch)
+        # train(train_loader, model, criterion, optimizer, epoch)
 
         # evaluate on validation set
         result, img_merge = validate(val_loader, model, epoch)