Fix train bug

2020-11-14 09:29:08 -07:00 · 2020-11-14 09:29:08 -07:00 · ec621c69b5
commit ec621c69b5
parent 9c3d0b7560
1 changed files with 1 additions and 1 deletions
--- 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()