From f62b7df1b23103fdad5f5e914b1e198eec8b5c18 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: