diff --git a/codes/models/steps/losses.py b/codes/models/steps/losses.py
index 878a60f5..28f692b7 100644
--- a/codes/models/steps/losses.py
+++ b/codes/models/steps/losses.py
@@ -134,7 +134,7 @@ class FeatureLoss(ConfigurableLoss):
         self.netF = define_F(which_model=opt['which_model_F'],
                              load_path=opt['load_path'] if 'load_path' in opt.keys() else None).to(self.env['device'])
         if not env['opt']['dist']:
-            self.netF = torch.nn.parallel.DataParallel(self.netF)
+            self.netF = torch.nn.parallel.DataParallel(self.netF, device_ids=env['opt']['gpu_ids'])
 
     def forward(self, _, state):
         with autocast(enabled=self.env['opt']['fp16']):
diff --git a/codes/train.py b/codes/train.py
index 27d34b47..3ecc1e04 100644
--- a/codes/train.py
+++ b/codes/train.py
@@ -221,7 +221,7 @@ class Trainer:
                         img_dir = os.path.join(opt['path']['val_images'], img_name)
                         util.mkdir(img_dir)
 
-                        self.model.feed_data(val_data)
+                        self.model.feed_data(val_data, self.current_step)
                         self.model.test()
 
                         visuals = self.model.get_current_visuals()