From 6c347282993d2e2db91b376d3113efa3774c3a22 Mon Sep 17 00:00:00 2001 From: junyanz Date: Sat, 22 Apr 2017 08:15:48 -0700 Subject: add dropout option for G --- options/base_options.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'options/base_options.py') diff --git a/options/base_options.py b/options/base_options.py index ec0f439..4074746 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -34,7 +34,7 @@ class BaseOptions(): self.parser.add_argument('--display_winsize', type=int, default=256, help='display window size') self.parser.add_argument('--display_id', type=int, default=1, help='window id of the web display') 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('--use_dropout', action='store_true', help='use dropout for the generator') self.initialized = True def parse(self): -- cgit v1.2.3-70-g09d2