From f53760bfd259d393da36ca49ac05622df214b8ce Mon Sep 17 00:00:00 2001
From: vieting <45091115+vieting@users.noreply.github.com>
Date: Thu, 27 Jul 2023 09:51:42 +0200
Subject: [PATCH] CompileTFGraphJob: allow Variable epoch (#435)

---
 returnn/compile.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/returnn/compile.py b/returnn/compile.py
index 567a5137..1310219b 100644
--- a/returnn/compile.py
+++ b/returnn/compile.py
@@ -19,7 +19,7 @@
 
 class CompileTFGraphJob(Job):
     """
-    This Job is a wrapper around the RETURNN tool comptile_tf_graph.py
+    This Job is a wrapper around the RETURNN tool compile_tf_graph.py
 
     """
 
@@ -47,7 +47,7 @@ def __init__(
         :param int train:
         :param int eval:
         :param int search:
-        :param int|None epoch: compile a specific epoch for networks that might change with every epoch
+        :param int|tk.Variable|None epoch: compile a specific epoch for networks that might change with every epoch
         :param int log_verbosity: RETURNN log verbosity from 1 (least verbose) to 5 (most verbose)
         :param str|None device: optimize graph for cpu or gpu. If `None`, defaults to cpu for current RETURNN.
             For any RETURNN version before `cd4bc382`, the behavior will depend on the `device` entry in the
@@ -115,7 +115,7 @@ def run(self):
         if self.device is not None:
             args.append("--device=%s" % self.device)
         if self.epoch is not None:
-            args.append("--epoch=%d" % self.epoch)
+            args.append("--epoch=%d" % util.instanciate_delayed(self.epoch))
         if self.summaries_tensor_name is not None:
             args.append("--summaries_tensor_name=%s" % self.summaries_tensor_name)
         if self.rec_step_by_step is not None: