From ee5d38e287e3dda25fa159baa7232d6f6588267a Mon Sep 17 00:00:00 2001 From: junyanz Date: Sat, 9 Sep 2017 02:42:21 -0700 Subject: update options --- options/base_options.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'options/base_options.py') diff --git a/options/base_options.py b/options/base_options.py index 3870048..d2ee44f 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -34,7 +34,6 @@ class BaseOptions(): self.parser.add_argument('--display_id', type=int, default=1, help='window id of the web display') self.parser.add_argument('--display_port', type=int, default=8097, help='visdom port of the web display') self.parser.add_argument('--display_single_pane_ncols', type=int, default=0, help='if positive, display all images in a single visdom web panel with certain number of images per row.') - 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('--no_dropout', action='store_true', help='no 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.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]') @@ -54,7 +53,7 @@ class BaseOptions(): id = int(str_id) if id >= 0: self.opt.gpu_ids.append(id) - + # set gpu ids if len(self.opt.gpu_ids) > 0: torch.cuda.set_device(self.opt.gpu_ids[0]) -- cgit v1.2.3-70-g09d2