进度条无用重复

时间:2019-01-24 15:10:39

标签: keras progress-bar

当我训练模型时,进度条是重复的,这让我有些烦恼。

我什么也没尝试,但是我还是不知道如何面对这个问题。这是我的屏幕上的样子:

enter image description here

这是我的火车发电机功能

def train_generator(
    self, data_gen, epochs, batch_size, steps_per_epoch, class_weights=None
):

    print("[Model] Training Started")
    print(
        "[Model] %s epochs, %s batch size, %s batches per epoch"
        % (epochs, batch_size, steps_per_epoch)
    )

    save_fname = "saved_models/%s-e%s.h5" % (
        dt.datetime.now().strftime("%d%m%Y-%H%M%S"),
        str(epochs),
    )
    callbacks = [
        ModelCheckpoint(filepath=save_fname, monitor="loss", save_best_only=True)
    ]

    self.model.fit_generator(
        data_gen,
        steps_per_epoch=steps_per_epoch,
        epochs=epochs,
        callbacks=callbacks,
        class_weight=class_weights,
    )

    print("[Model] Training Completed. Model saved as %s" % save_fname)

如何摆脱重复和无用的进度条?

0 个答案:

没有答案