diff --git a/train.py b/train.py
index 24efefa98653..1e88c25f3606 100644
--- 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__":