代码之家  ›  专栏  ›  技术社区  ›  Jan Slominski

Keras 2 fit\u generator用户警告:“steps\u per\u epoch”与Keras 1参数“samples\u per\u epoch”不同`

  •  6
  • Jan Slominski  · 技术社区  · 9 年前

    每当我使用fit_generator()时,我都会收到以下警告:

    主要的 .py:5:UserWarning:Keras 2参数的语义 steps_per_epoch 与Keras 1参数不同 samples_per_epoch . 是每个历元从生成器中提取的批次数。基本上,steps\u per\u epoch=samples\u per\u epoch/batch\u size。类似地 nb_val_samples -&燃气轮机; validation_steps 和 val_samples -&燃气轮机; steps 争论已经改变。相应地更新方法调用。 主要的 .py:5:用户警告:更新您的 fit_generator 调用Keras 2 API: fit_generator(<keras.pre..., steps_per_epoch=60000, validation_data=<keras.pre..., epochs=1, validation_steps=10000)

    以下是我的模型代码(简单的MNIST线性分类器,但我使用的每个模型都会收到此警告):

    model = Sequential([
        Lambda(normalize_input, input_shape=(1, 28, 28)),
        Flatten(),
        Dense(10, activation='softmax')
    ])
    model.compile(Adam(),
                  loss='categorical_crossentropy',
                  metrics=['accuracy'])
    

    model.fit_generator(batches, 
                     steps_per_epoch=steps_per_epoch, 
                     nb_epoch=1, 
                     validation_data=test_batches, 
                     nb_val_samples=test_batches.n)
    

    我明白这个警告告诉我什么。这对我来说不是问题。我怎样才能摆脱它?

    1 回复  |  直到 9 年前
        1
  •  12
  •   Yu-Yang    9 年前

    如果函数调用中存在任何Keras 1.0关键字,则会出现此警告。通过替换来更新函数调用 nb_epoch 具有 epochs 和 nb_val_samples validation_steps .

    推荐文章