Update run.py

Author: 2KangHo

run.py

@@ -3,6 +3,7 @@
 import argparse
 import torch
 import torch.nn as nn
+import time
 import math
 from torch.autograd import Variable
 from PIL import Image
@@ -46,7 +47,10 @@
 if opt.cuda:
     input = input.cuda()
 
+start_time = time.time()
 out = model(input)
+elapsed_time = time.time() - start_time
+print("===> It takes {:.4f} seconds.".format(elapsed_time))
 out = out.cpu()
 
 print("type = ", type(out))