diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-05-02 20:26:39 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-05-02 20:26:39 -0700 |
| commit | 349614a2f168654ba59bf1461ea61e1cb9358eb6 (patch) | |
| tree | be9829470eb5dddc673b9019f552a1292f2e3ecf /options/base_options.py | |
| parent | e9146a659a896fc71ebf59cc57248449091e8d16 (diff) | |
use InstanceNorm2d from the latest pytorch
Diffstat (limited to 'options/base_options.py')
| -rw-r--r-- | options/base_options.py | 2 |
1 files changed, 1 insertions, 1 deletions
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') |