From 011545193f1fc55975956d25a48f568ab426286b Mon Sep 17 00:00:00 2001
From: Fize Jacques <jacques.fize@cirad.fr>
Date: Thu, 5 Nov 2020 09:34:37 +0100
Subject: [PATCH] little UPD
---
train_geocoder_v2.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/train_geocoder_v2.py b/train_geocoder_v2.py
index e67f311..4a51665 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],
--
GitLab