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

Disabled early stopping

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-09 11:27:45 +02:00
parent 320ba40389
commit 0c7e96ee9d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -491,7 +491,7 @@ def train_keras(train_generator: callable,
save_weights_only=False save_weights_only=False
), ),
tf.keras.callbacks.TerminateOnNaN(), tf.keras.callbacks.TerminateOnNaN(),
tf.keras.callbacks.EarlyStopping(patience=2, min_delta=0.001, monitor="val_loss") # tf.keras.callbacks.EarlyStopping(patience=2, min_delta=0.001, monitor="val_loss")
] ]
if tensorboard_callback is not None: if tensorboard_callback is not None:
callbacks.append(tensorboard_callback) callbacks.append(tensorboard_callback)