fixed bug that was making parser pick up unkown args from the environment making it crash. Eg in colab it was picking up jupyter runtime dir in root location. Now we don't need the dummy variable -f in colab

This commit is contained in:
Aswani committed 2021-07-24 00:03:20 -07:00
1 parent ab198b253d
commit 5e0c4933a6
2 files changed
+573 -574

No files matched your search

+1 -1
View File
@@ -69,7 +69,7 @@ class BaseOptions():
def parse(self, save=True):
if not self.initialized:
self.initialize()
self.opt = self.parser.parse_args()
self.opt, unknown_arg = self.parser.parse_known_args()
self.opt.isTrain = self.isTrain # train or test
str_ids = self.opt.gpu_ids.split(',')