python - 如何加载用 tf.keras.models.save_model 保存的训练模型的权重？

Question

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

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

我尝试使用 new_model = tf.keras.models.load_model(dirpath) 加载模型，但它给出了一个 ValueError 因为我使用的是自定义模型(看起来，我创建了一个类继承自 tf.keras.Model)。所以我尝试实例化一个新模型，然后使用

加载权重

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

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

OSError: Unable to open file (unable to open file: name = 'dirpath/variables', errno = 13, error message = 'Permission denied', flags = 0, o_flags = 0)

那么如何将权重加载到模型上呢？文件在那里，我只是不知道如何将它们放回我的模型中。

Answer 1

弄清楚了，我用错了路径。我需要执行 model.load_weights(dirpath/variables/variables)。有两个名为 variables 的文件，具有不同的扩展名(.data-00000-of-00001 和 .index)，这就是名称你想打电话。