a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Fix train bug

Browse Source
This commit is contained in:
James Betker 2020-11-14 09:29:08 -07:00
parent 9c3d0b7560
commit ec621c69b5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/train.py
View File

@ -221,7 +221,7 @@ class Trainer:
img_dir = os.path.join(opt['path']['val_images'], img_name) img_dir = os.path.join(opt['path']['val_images'], img_name)
util.mkdir(img_dir) util.mkdir(img_dir)
self.model.feed_data(val_data) self.model.feed_data(val_data, self.current_step)
self.model.test() self.model.test()
visuals = self.model.get_current_visuals() visuals = self.model.get_current_visuals()