From 9cf80b7f6098391f70efa01b4fd156143752ef47 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Christoph=20F=C3=BCrbacher?=
 <91476135+chf4850@users.noreply.github.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__":