diff options
| -rwxr-xr-x | options/base_options.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/options/base_options.py b/options/base_options.py index 561a890..0b7444a 100755 --- a/options/base_options.py +++ b/options/base_options.py @@ -63,7 +63,8 @@ class BaseOptions(): def parse(self, save=True): if not self.initialized: self.initialize() - self.opt = self.parser.parse_args() + self.opt, unknown = self.parser.parse_known_args() + self.opt.unknown = unknown self.opt.isTrain = self.isTrain # train or test str_ids = self.opt.gpu_ids.split(',') |