From 349614a2f168654ba59bf1461ea61e1cb9358eb6 Mon Sep 17 00:00:00 2001 From: junyanz Date: Tue, 2 May 2017 20:26:39 -0700 Subject: use InstanceNorm2d from the latest pytorch --- 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 cce6aae..9ec7c9a 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -29,7 +29,7 @@ class BaseOptions(): self.parser.add_argument('--which_direction', type=str, default='AtoB', help='AtoB or BtoA') self.parser.add_argument('--nThreads', default=2, type=int, help='# threads for loading data') self.parser.add_argument('--checkpoints_dir', type=str, default='./checkpoints', help='models are saved here') - self.parser.add_argument('--norm', type=str, default='batch', help='batch normalization or instance normalization') + self.parser.add_argument('--norm', type=str, default='instance', help='instance normalization or batch normalization') self.parser.add_argument('--serial_batches', action='store_true', help='if true, takes images in order to make batches, otherwise takes them randomly') 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') -- cgit v1.2.3-70-g09d2