2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Change to Keras optimizer

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-04 12:33:34 +02:00
parent d9b742f61a
commit 59a1d85af9
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/twomartens/masterthesis/ssd.py
View File

@ -471,8 +471,11 @@ 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.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),
loss=ssd_loss.compute_loss,
metrics=[
"categorical_accuracy"