@@ -1091,7 +1091,7 @@ def generate_arrays_from_file(path):
1091
1091
1092
1092
# start generator thread storing batches into a queue
1093
1093
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 )
1095
1095
if do_validation and not val_gen :
1096
1096
X_val , y_val , sample_weight_val = self ._check_generator_output (validation_data ,
1097
1097
_data_stop )
@@ -1142,7 +1142,8 @@ def generate_arrays_from_file(path):
1142
1142
val_outs = self .evaluate_generator (validation_data ,
1143
1143
nb_val_samples ,
1144
1144
show_accuracy = show_accuracy ,
1145
- verbose = 0 )
1145
+ verbose = 0 , nb_worker = nb_val_worker ,
1146
+ wait_time = wait_time )
1146
1147
else :
1147
1148
val_outs = self .evaluate (X_val , y_val ,
1148
1149
show_accuracy = show_accuracy ,
@@ -1600,7 +1601,7 @@ def generate_arrays_from_file(path):
1600
1601
1601
1602
# start generator thread storing batches into a queue
1602
1603
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 )
1604
1605
if do_validation and not val_gen :
1605
1606
# TODO: _data_stop not really sensical here
1606
1607
data_val , sample_weight_val = self ._check_generator_output (validation_data , _data_stop )
@@ -1649,7 +1650,9 @@ def generate_arrays_from_file(path):
1649
1650
if val_gen :
1650
1651
val_outs = self .evaluate_generator (validation_data ,
1651
1652
nb_val_samples ,
1652
- verbose = 0 )
1653
+ verbose = 0 ,
1654
+ nb_worker = nb_val_worker ,
1655
+ wait_time = wait_time )
1653
1656
else :
1654
1657
val_outs = self .evaluate (data_val ,
1655
1658
sample_weight = sample_weight_val ,
0 commit comments