Question

我正在建立图像分类器。

具有我想要的输出指标。

我添加了ImageDataGenerator，并且验证步骤在主输出的中间添加了自己的输出指标。像这样：

Epoch 58/300
10000/10000 [==============================] - 1s 65us/sample - loss: 0.7600 - acc: 0.8301
 - 15s - loss: 0.5111 - acc: 0.9054 - val_loss: 0.7654 - val_acc: 0.8301

这是模型编译器：

model.compile(
    optimizer= SGD(
        lr= 0.0,
        momentum= hp['momentum'],
        nesterov= True,
        ),
    loss= categorical_crossentropy,
    metrics= ['accuracy'],
    )

这是ImageDataGenerator：

datagen = ImageDataGenerator(
    width_shift_range=4,
    height_shift_range=4,
    fill_mode= 'constant',
    cval = 0.0,
    horizontal_flip=True,
    )

在添加生成器之前，这里是fit函数调用：

model.fit(
    x= train_data,
    y= train_labels,
    epochs= hp['num_epoch'],
    verbose= 2,
    callbacks= [lr_scheduler],
    validation_data= (
        valid_data, valid_labels
        ),
    )

Epoch 8/300
- 21s - loss: 0.8771 - acc: 0.8077 - val_loss: 1.1955 - val_acc: 0.7258

然后之后生成器：

model.fit_generator(
    datagen.flow(
        train_data, train_labels,
        batch_size= hp['batch_size'],
        ),
    epochs= hp['num_epoch'],
    verbose= 2,
    callbacks= [lr_scheduler],
    validation_data= (
        valid_data, valid_labels),
    )

Epoch 58/300
10000/10000 [==============================] - 1s 65us/sample - loss: 0.7600 - acc: 0.8301
 - 15s - loss: 0.5111 - acc: 0.9054 - val_loss: 0.7654 - val_acc: 0.8301

这是一个已知的错误吗？我应该提交错误报告吗？我做傻了吗？（希望）有一个简单的解决方案吗？

谢谢，我们将不胜感激。

Answer 1

我从TF 1.13.0更新到TF 1.14.0，并且清除了。

显然TensorFlow或Keras已发现此错误。

验证步骤冗长

1 个答案: