From cf0510d147964f9407a2c0225688d6c036549b22 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Wed, 25 Dec 2019 20:28:15 -0400 Subject: [PATCH] next --- cyclegan/test.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/cyclegan/test.py b/cyclegan/test.py index 1caabeb..916d5bb 100644 --- a/cyclegan/test.py +++ b/cyclegan/test.py @@ -84,6 +84,11 @@ if __name__ == '__main__': if i % 5 == 0: # save images to an HTML file print('processing (%04d)-th image... %s' % (i, img_path)) save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio, width=opt.display_winsize) + + model.fake = None + model.real = None + model.fake_noattack = None + webpage.save() # save the HTML