From eadd4a927378c89a6345e581431396731e035aa4 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 5 Jun 2019 12:22:22 +0200 Subject: [PATCH] Added missing keyword argument Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 514a2e1..7266472 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -444,7 +444,8 @@ def _train_one_epoch(epoch: int, ssd: tf.keras.Model, ssd_optimizer: tf.train.Optimizer, global_step: tf.Variable, - epoch_var: tf.Variable) -> Dict[str, float]: + epoch_var: tf.Variable, + learning_rate_var: tf.Variable) -> Dict[str, float]: with summary_ops_v2.always_record_summaries(): epoch_var.assign(epoch)