Minor bug fixes

code/data/demo.py

@@ -15,6 +15,7 @@ class Demo(data.Dataset):
         self.scale = args.scale
         self.idx_scale = 0
         self.train = False
+        self.benchmark = False
 
         self.filelist = []
         for f in os.listdir(args.dir_demo):

code/trainer.py

@@ -49,7 +49,7 @@ class Trainer():
             timer_model.tic()
 
             self.optimizer.zero_grad()
-            sr = self.model(lr, self.args.scale[idx_scale])
+            sr = self.model(lr, idx_scale)
             loss = self.loss(sr, hr)
             if loss.data[0] < self.args.skip_threshold * self.error_last:
                 loss.backward()