From e5b2fd6d36b4297c4314478e88820cd10943d192 Mon Sep 17 00:00:00 2001 From: Taesung Park Date: Thu, 27 Apr 2017 01:28:53 -0700 Subject: 1. Added one_direction_test_model that generates the outputs in only one direction 2. Changed the option naming from ntrain to max_dataset_size --- options/base_options.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'options/base_options.py') diff --git a/options/base_options.py b/options/base_options.py index f03a687..bce0b9c 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -35,6 +35,8 @@ class BaseOptions(): 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.parser.add_argument('--max_dataset_size', type=int, default=float("inf"), help='Maximum number of samples allowed per dataset. If the dataset directory contains more than max_dataset_size, only a subset is loaded.') + self.initialized = True def parse(self): -- cgit v1.2.3-70-g09d2