此代码不适用于tensorflow 2.16.0+版本

Question

checkpoint = ModelCheckpoint(
    './base.model',
    monitor='val_accuracy',
    verbose=1,
    save_best_only=True,
    mode='max',
    save_weights_only=False,
    save_frequency=1
)
earlystop = EarlyStopping(
    monitor='val_loss',
    min_delta=0.001,
    patience=30,
    verbose=1,
    mode='auto'
)

opt1 = tf.keras.optimizers.Adam()

callbacks = [checkpoint,earlystop]

这不适用于张量流 2.16.1 但是，正在 google colab 上开发 2.15.0

我如何修复我的代码或如何安装tensorflow 2.15.0？

我试过了

pip install tensorflow=2.15.0

但是，它显示错误

Answer 1

您评论中的错误表明

save_frequency

是

ModelCheckpoint

的未知关键字。 在文档中显示

ModelCheckpoint

期望

save_freq

作为参数。另请注意，如果您为

save_freq

设置一个整数，它会保存每个 x 批次，而不是纪元。如果你只是想让它保存每个纪元，你可以设置

save_freq='epoch'

这也是默认的。 为什么会在 2.15 中醒来，我不能说，2.15 的文档中也有

save_freq

。也许到目前为止它与后台的

save_frequency

兼容。最好使用

save_freq

，因为它是官方支持的。

如果您想更改 colab 中的软件包版本，请使用

%pip install -U "tensorflow~=2.15.0"

%

在 python 环境之外启用 pip 命令，

-U

升级（或在您的情况下降级）现有版本，并且您需要

""

以及 == 或 ~= 作为版本号。 == 代表这个版本，~= 代表，在本例中为“2.15.x”，其中 x 是最新版本。