fanwuyang/neural-enhance-super-resolution
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added --save-every-epoch option

Browse Source
This commit is contained in:
Tom White
2016-10-31 00:12:28 +13:00
parent f68f04fb1c
commit b05ee6ad08
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enhance.py
+3
View File
@@ -39,6 +39,7 @@ add_arg('--scales', default=2, type=int, help='How ma
add_arg('--model', default='medium', type=str, help='Name of the neural network to load/save.') add_arg('--model', default='medium', type=str, help='Name of the neural network to load/save.')
add_arg('--model', default='small', type=str, help='Name of the neural network to load/save.') add_arg('--model', default='small', type=str, help='Name of the neural network to load/save.')
add_arg('--train', default=False, type=str, help='File pattern to load for training.') add_arg('--train', default=False, type=str, help='File pattern to load for training.')
add_arg('--save-every-epoch', default=False, action='store_true', help='Save generator after every training epoch.')
add_arg('--batch-resolution', default=192, type=int, help='Resolution of images in training batch.') add_arg('--batch-resolution', default=192, type=int, help='Resolution of images in training batch.')
add_arg('--batch-size', default=15, type=int, help='Number of images per training batch.') add_arg('--batch-size', default=15, type=int, help='Number of images per training batch.')
add_arg('--buffer-size', default=1500, type=int, help='Total image fragments kept in cache.') add_arg('--buffer-size', default=1500, type=int, help='Total image fragments kept in cache.')
@@ -484,6 +485,8 @@ class NeuralEnhancer(object):
print(' - adversary mode: generator engaging discriminator.') print(' - adversary mode: generator engaging discriminator.')
self.model.adversary_weight.set_value(args.adversary_weight) self.model.adversary_weight.set_value(args.adversary_weight)
running = None running = None
if args.save_every_epoch:
self.model.save_generator()
except KeyboardInterrupt: except KeyboardInterrupt:
pass pass