如何加载训练模型的权重，这些权重是用 tf.keras.models.save_model 保存的？

Question

我训练了一个神经网络，没有任何检查点，最后我写了tf.keras.models.save_model(model, dirpath)来保存整个 model，它创建了以下文件：

savedmodel.pb 
assets/
variables/variables.index
variables/variables.data-00000-of-00001

I tried loading the model using new_model = tf.keras.models.load_model(dirpath) , but it gave a ValueError because I'm using a custom model (it seems, I created a class inheriting from tf.keras.Model ). 因此，我尝试实例化一个新的 model，然后使用

model = myModel(someArgs)
model.load_weights(dirpath/variables)

但是，我收到以下错误消息：

OSError：无法打开文件（无法打开文件：name = 'dirpath/variables'，errno = 13，错误消息 = 'Permission denied'，flags = 0，o_flags = 0）

那么如何将砝码加载到 model 上？ 文件在那里，我只是不知道如何将它们放回我的 model 中。

Answer 1

想通了，我使用了错误的路径。 我需要做model.load_weights(dirpath/variables/variables) 。 吃了两个名为variables的文件，具有不同的扩展名（ .data-00000-of-00001和.index ），这就是您要调用的名称。