Switch default to small model to reduce memory usage.

main
Alex J. Champandard 9 years ago
parent 2b5fc8f51d
commit 203917d122

@ -37,6 +37,7 @@ add_arg = parser.add_argument
add_arg('files', nargs='*', default=[]) add_arg('files', nargs='*', default=[])
add_arg('--scales', default=2, type=int, help='How many times to perform 2x upsampling.') add_arg('--scales', default=2, type=int, help='How many times to perform 2x upsampling.')
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('--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('--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.')

Loading…
Cancel
Save