From 2b93480df9f612a363ef1f3b5dbd13eed763f9de Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 4 Jul 2019 12:59:39 +0200 Subject: [PATCH] Reversed optimizer to Tensorflow and changed beta1 to 0.9 Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 3b6fbe8..f4287bb 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -471,11 +471,8 @@ def train_keras(train_generator: callable, # compile the model ssd_model.model.compile( - # optimizer=tf.train.AdamOptimizer(learning_rate=learning_rate_var, - # beta1=0.5, beta2=0.999), - optimizer=tf.keras.optimizers.Adam(lr=learning_rate_var, - beta_1=0.9, beta_2=0.999, - epsilon=1e-08, decay=5e-04), + optimizer=tf.train.AdamOptimizer(learning_rate=learning_rate_var, + beta1=0.9, beta2=0.999), loss=ssd_loss.compute_loss, metrics=[ "categorical_accuracy"