return opt from train.run() (#6581)

SecretStar112 · Feb 9, 2022 · beded5b · beded5b
1 parent 043f62e
commit beded5b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -634,6 +634,7 @@ def run(**kwargs):
     for k, v in kwargs.items():
         setattr(opt, k, v)
     main(opt)
+    return opt
 
 
 if __name__ == "__main__":