拟合模型时仅显示每个时期的摘要,而不显示每个步骤

时间:2018-12-04 22:13:46

标签: tensorflow

是否可以在控制台的时代打印运行摘要。

对于初学者来说,我希望在纪元的开始和结束时都能看到损失和准确性。

当前,它将在每个时期打印信息。我认为这将提高性能,并且我仍然能够确定我的学习率设置是否正确。

编辑: 与代码相关

    data, labels = self.FixInput(data,labels)
    v_data, v_labels = self.FixInput(v_data,v_labels)
    opt = SGD(lr=learningRate)

    self.Model.compile(loss='categorical_crossentropy',
                  optimizer=opt,
                  metrics=['accuracy'])

    tb = TensorBoard(log_dir="logs/{}".format(time.time()))

    historydata = self.Model.fit(data, labels, epochs=10, steps_per_epoch=600, callbacks=[tb],
              validation_data=(v_data, v_labels), validation_steps=400)

    return historydata.history

0 个答案:

没有答案