From 832ffb7bcfbc00eff869c6cdc03f04c265dcae8d Mon Sep 17 00:00:00 2001
From: Albert Zeyer <albzey@gmail.com>
Date: Wed, 29 Nov 2023 10:51:08 +0100
Subject: [PATCH] config startup_callback

---
 returnn/__main__.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/returnn/__main__.py b/returnn/__main__.py
index 8c41f8a2d6..bd5351ff52 100644
--- 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()