From beded5b837706e6dffd025487f8389b58cfb0c7f Mon Sep 17 00:00:00 2001
From: SecretStar112 <edwardxian062@gmail.com>
Date: Wed, 9 Feb 2022 15:11:52 +0100
Subject: [PATCH] return `opt` from `train.run()` (#6581)

---
 train.py | 1 +
 1 file changed, 1 insertion(+)

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