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