什么相当于tensorflow中的torch.load()？

【问题标题】：what is equivalent to torch.load() in tensorflow?什么相当于tensorflow中的torch.load()？
【发布时间】：2021-12-17 21:07:08
【问题描述】：

我想知道有没有办法在tensorflow中查看模型的参数。 pytorch 中有一个命令，即 torch.load('/filepath')。

【问题讨论】：

【解决方案1】：

对于预测上下文，你可以做一个

model = tf.keras.models.load_model(PATH, compile=True)

这适用于 .h5 keras 模型和 SavedModel 格式的模型。否则，您可能必须提供预测上下文中可能没有的自定义指标和训练代码。

【讨论】：

【解决方案2】：

假设您已经有一个模型保存在MODEL_PATH，这应该可以解决问题：

model = tf.keras.models.load_model(MODEL_PATH)
model.summary()

Check this out 了解有关保存和加载模型的更多信息。

【讨论】：