Skip to content

S
seminar-breakout
Commit 2d07bc0a authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

reset the cyclegan validation dataset before use (fix #411)

parent 52664b97
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/GAN/CycleGAN.py
View file @ 2d07bc0a
...@@ -193,6 +193,7 @@ class VisualizeTestSet(Callback): ...@@ -193,6 +193,7 @@ class VisualizeTestSet(Callback):
def _before_train(self): def _before_train(self):
global args global args
self.val_ds = get_data(args.data, isTrain=False) self.val_ds = get_data(args.data, isTrain=False)
self.val_ds.reset_state()
def _trigger(self): def _trigger(self):
idx = 0 idx = 0
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment