add RETURNN version to learning rates file

rwth-i6 · Apr 5, 2024 · 0d67dc6 · 0d67dc6
1 parent 3f78487
commit 0d67dc6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/returnn/torch/engine.py b/returnn/torch/engine.py
@@ -284,6 +284,7 @@ def init_train_epoch(self):
             {
                 "global_train_step": self.global_train_step,
                 "effective_learning_rate": self._updater.get_effective_learning_rate(),
+                "returnn": util.describe_returnn_version(),
                 "torch": util.describe_torch_version(),
                 "time": time.strftime("%Y-%m-%d-%H-%M-%S (UTC%z)"),
                 "hostname": socket.gethostname(),