From 7800d516596f1a25986b458cddf8b8785bcc7df8 Mon Sep 17 00:00:00 2001 From: junyanz Date: Fri, 6 Oct 2017 10:46:43 -0700 Subject: support nc=1, add new leaerning rate policy and new initialization --- options/base_options.py | 1 + 1 file changed, 1 insertion(+) (limited to 'options/base_options.py') diff --git a/options/base_options.py b/options/base_options.py index de8bc74..c1b0733 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -39,6 +39,7 @@ class BaseOptions(): self.parser.add_argument('--resize_or_crop', type=str, default='resize_and_crop', help='scaling and cropping of images at load time [resize_and_crop|crop|scale_width|scale_width_and_crop]') self.parser.add_argument('--no_flip', action='store_true', help='if specified, do not flip the images for data augmentation') self.parser.add_argument('--identity', type=float, default=0.0, help='use identity mapping. Setting identity other than 1 has an effect of scaling the weight of the identity mapping loss. For example, if the weight of the identity loss should be 10 times smaller than the weight of the reconstruction loss, please set optidentity = 0.1') + self.parser.add_argument('--init_type', type=str, default='xavier', help='network initialization [normal|xavier|kaiming|orthogonal]') self.initialized = True -- cgit v1.2.3-70-g09d2