Skip to content

Commit ca05efc

Browse files
committed
Merge pull request keras-team#1862 from qdbp/issue_1857
Fixing issue 1857
2 parents 0daec53 + 7768ae0 commit ca05efc

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

keras/models.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1091,7 +1091,7 @@ def generate_arrays_from_file(path):
10911091

10921092
# start generator thread storing batches into a queue
10931093
data_gen_queue, _data_stop = generator_queue(generator, max_q_size=max_data_q_size,
1094-
wait_time=wait_time)
1094+
wait_time=wait_time, nb_worker=nb_worker)
10951095
if do_validation and not val_gen:
10961096
X_val, y_val, sample_weight_val = self._check_generator_output(validation_data,
10971097
_data_stop)
@@ -1142,7 +1142,8 @@ def generate_arrays_from_file(path):
11421142
val_outs = self.evaluate_generator(validation_data,
11431143
nb_val_samples,
11441144
show_accuracy=show_accuracy,
1145-
verbose=0)
1145+
verbose=0, nb_worker=nb_val_worker,
1146+
wait_time=wait_time)
11461147
else:
11471148
val_outs = self.evaluate(X_val, y_val,
11481149
show_accuracy=show_accuracy,
@@ -1600,7 +1601,7 @@ def generate_arrays_from_file(path):
16001601

16011602
# start generator thread storing batches into a queue
16021603
data_gen_queue, _data_stop = generator_queue(generator, max_q_size=max_data_q_size,
1603-
wait_time=wait_time)
1604+
wait_time=wait_time, nb_worker=nb_worker)
16041605
if do_validation and not val_gen:
16051606
# TODO: _data_stop not really sensical here
16061607
data_val, sample_weight_val = self._check_generator_output(validation_data, _data_stop)
@@ -1649,7 +1650,9 @@ def generate_arrays_from_file(path):
16491650
if val_gen:
16501651
val_outs = self.evaluate_generator(validation_data,
16511652
nb_val_samples,
1652-
verbose=0)
1653+
verbose=0,
1654+
nb_worker=nb_val_worker,
1655+
wait_time=wait_time)
16531656
else:
16541657
val_outs = self.evaluate(data_val,
16551658
sample_weight=sample_weight_val,

0 commit comments

Comments
 (0)