config startup_callback

rwth-i6 · Nov 29, 2023 · 832ffb7 · 832ffb7
1 parent 93236a2
commit 832ffb7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/returnn/__main__.py b/returnn/__main__.py
@@ -394,6 +394,9 @@ def init(config_filename=None, command_line_options=(), config_updates=None, ext
     init_backend_engine()
     if config.bool("ipython", False):
         debug_util.init_ipython_kernel()
+    if config.typed_value("startup_callback"):
+        startup_callback = config.typed_value("startup_callback")
+        startup_callback(config=config)
     if need_data():
         init_data()
     print_task_properties()