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
2021-07-24 00:03:20 -07:00
parent ab198b253d
commit 5e0c4933a6
2 changed files with 573 additions and 574 deletions
+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(',')