Skip to content
Snippets Groups Projects
Commit d21f632d authored by Ludovic Moncla's avatar Ludovic Moncla
Browse files

Update train_geocoder.py

parent 99cf3e26
No related branches found
No related tags found
No related merge requests found
......@@ -203,7 +203,7 @@ print(model.summary())
checkpoint = ModelCheckpoint(MODEL_OUTPUT_FN + ".part", monitor='loss', verbose=1,
save_best_only=True, mode='auto', period=1)
epoch_timer = EpochTimer("outputs/"+PREFIX_OUTPUT_FN+"_epoch_timer_output.csv")
epoch_timer = EpochTimer(HISTORY_FN)
......@@ -213,9 +213,6 @@ history = model.fit(training_generator,verbose=True,
hist_df = pd.DataFrame(history.history)
hist_df.to_csv(HISTORY_FN)
model.save(MODEL_OUTPUT_FN)
# Erase Model Checkpoint file
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment