args

2 files changed, 7 insertions, 3 deletions

diff --git a/options/base_options.py b/options/base_options.py
index 814ea3e..f741927 100644
--- a/options/base_options.py
+++ b/options/base_options.py
@@ -48,7 +48,8 @@ class BaseOptions():
     def parse(self):
         if not self.initialized:
             self.initialize()
-        self.opt = self.parser.parse_known_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(',')
diff --git a/options/dataset_options.py b/options/dataset_options.py
index 5eca374..fcb9654 100644
--- a/options/dataset_options.py
+++ b/options/dataset_options.py
@@ -178,8 +178,11 @@ class DatasetOptions(BaseOptions):
           help='canny high threshold'
         )
 
-    def parse(self):
+    def parse(self, args=None):
         if not self.initialized:
             self.initialize()
-        self.opt = self.parser.parse_known_args()
+        if args is not None:
+          self.opt, unknown = self.parser.parse_known_args(args)
+        else:
+          self.opt, unknown = self.parser.parse_known_args()
         return self.opt
\ No newline at end of file