keras 如何加载检查点

dkqlctbz  于 2022-11-13  发布在  其他
关注(0)|答案(1)|浏览(127)

您好,我已尝试加载检查点,但收到以下错误:
“无法打开../代码输出/带旋转的三维新架构:数据丢失:不是stable(错误的幻数):也许您的文件是不同的文件格式,您需要使用不同的还原操作符?”
这是我使用的代码:

checkpoint_filepath = '../codeOutputs/3DNewArchitectureWithRotation'
model_checkpoint_callback = tf.keras.callbacks.ModelCheckpoint(
    filepath=checkpoint_filepath,
    monitor='val_loss',
    verbose=0,
    save_best_only=False,
    save_weights_only=False,
    mode='auto',
    save_freq='epoch',
    options=None,
    initial_value_threshold=None,
   
)    



Model.load_weights(checkpoint_filepath)

BestRegressor = Model.fit(aaaiTrainImages, afTrainPorosity, validation_data = (aaaiValidationImages, afValidationPorosity), epochs=Epochs, callbacks =[EarlyStop,model_checkpoint_callback], verbose=2)

似乎检查点保存的文件类型是:HDF文档(应用程序/x-hdf)。我将感谢任何帮助,因为我花了很多天训练我的模型,突然崩溃,所以如果我可以跳过重新训练它到我的数据,这将是非常有帮助的

xxls0lw8

xxls0lw81#

我也遇到了同样的问题。正如其他人所指出的,这个问题源于参数save_weights_only=False,它创建了一个文件目录。你仍然可以调用model.load_weights()并取消持久化模型,但是你会得到那个令人不快的错误。我采取的一个方法是使用下面的方法取消持久化模型,而没有任何错误/警告。

import tensorflow as tf

m = tf.keras.models.load_model('/path/to/checkpoint/dir')

相关问题