diff --git a/train_geocoder.py b/train_geocoder.py index 4b7bfc3c94a20e19b44cac248b98538c44774ce2..c2c15e881ebef7c084305077229d1c9d1385671d 100644 --- a/train_geocoder.py +++ b/train_geocoder.py @@ -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