|
|
@@ -32,10 +32,10 @@ cmd:option("-random_overlay_rate", 0.0, 'data augmentation using flipped image o
|
|
|
cmd:option("-random_half_rate", 0.0, 'data augmentation using half resolution image (0.0-1.0)')
|
|
|
cmd:option("-random_unsharp_mask_rate", 0.0, 'data augmentation using unsharp mask (0.0-1.0)')
|
|
|
cmd:option("-scale", 2.0, 'scale factor (2)')
|
|
|
-cmd:option("-learning_rate", 0.001, 'learning rate for adam')
|
|
|
+cmd:option("-learning_rate", 0.0005, 'learning rate for adam')
|
|
|
cmd:option("-crop_size", 46, 'crop size')
|
|
|
cmd:option("-max_size", 256, 'if image is larger than max_size, image will be crop to max_size randomly')
|
|
|
-cmd:option("-batch_size", 32, 'mini batch size')
|
|
|
+cmd:option("-batch_size", 8, 'mini batch size')
|
|
|
cmd:option("-patches", 16, 'number of patch samples')
|
|
|
cmd:option("-inner_epoch", 4, 'number of inner epochs')
|
|
|
cmd:option("-epoch", 30, 'number of epochs to run')
|
nagadomi