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

chf4850 · web-flow · commit 108d58cbe6ae · 2022-02-09T15:11:52.000+01:00
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__":
