robocars/robocar-training
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix refactor from aws execution logs

Browse Source
This commit is contained in:
Cyrille Nofficial 2020-03-02 19:20:42 +01:00
parent 2076b4491a
commit 3a376dd5a3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tf_container/train_entry_point.py
View File

@ -77,9 +77,9 @@ def train():
return arr
logs = callbacks.TensorBoard(log_dir='logs', histogram_freq=0, write_graph=True, write_images=True)
save_best = callbacks.ModelCheckpoint('/opt/ml/model/model_cat', monitor='angle_out_loss', verbose=1,
save_best = callbacks.ModelCheckpoint('/opt/ml/model/model_cat', monitor='val_loss', verbose=1,
save_best_only=True, mode='min')
early_stop = callbacks.EarlyStopping(monitor='angle_out_loss',
early_stop = callbacks.EarlyStopping(monitor='val_loss',
min_delta=.0005,
patience=10,
verbose=1,