tensorflow keras评估函数运行时间过长

Question

我尝试在Google Colab中使用model.evaluate，代码如下：

test_datagen = ImageDataGenerator(rescale = 1./255.)
test_generator = test_datagen.flow_from_directory(test_dir,
batch_size = 20,
class_mode = 'binary', 
target_size = (image_size, image_size))
results = model.evaluate(test_generator)
for name, value in zip(model.metrics_names, results):
print(name, ': ', value)

我在控制台中得到以下输出：找到属于2个类别的624个图像。我期望模型.evaluate在ceil(624/20(=32步内完成。相反，该功能似乎永远在运行(远远超过1000步(。有人知道我做错了什么吗？

Answer 1

我最终指定了步骤参数

results = model.evaluate(test_generator, steps=len(test_generator))

len(test_generator(等于预期的32