diff --git a/train_geocoder_v2.py b/train_geocoder_v2.py
index e67f31173f8fd905857c591c11ad9f17d7ac130d..4a516656ec15089de8a9ee506cee6adebc1ff663 100644
--- a/train_geocoder_v2.py
+++ b/train_geocoder_v2.py
@@ -207,10 +207,8 @@ epoch_timer = EpochTimer("outputs/"+PREFIX_OUTPUT_FN+"_epoch_timer_output.csv")
 
 
 
-history = model.fit_generator(generator=training_generator,
+history = model.fit(training_generator,verbose=True,
                     validation_data=validation_generator,
-                    use_multiprocessing=True,
-                    workers=6,
                     callbacks=[checkpoint,epoch_timer],epochs=EPOCHS)
 
 # history = model.fit(x=[X_1_train,X_2_train],