diff options
Diffstat (limited to 'options/base_options.py')
| -rw-r--r-- | options/base_options.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/options/base_options.py b/options/base_options.py index 5bdd85e..150174b 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -49,6 +49,7 @@ class BaseOptions(): self.initialize() self.opt = self.parser.parse_args() self.opt.isTrain = self.isTrain # train or test + return self.opt str_ids = self.opt.gpu_ids.split(',') self.opt.gpu_ids = [] |