我发现TensorFlow支持各种模型文件.ckpt, .meta, .pb
1)现在我有一个resnet50模型,.ckpt
102MB和.pb
459KB,如何使用TensorFlow将它们加载到内存中?
2)Tensorflow支持多少种模型文件?谁能提供类似的清单或材料(例如1. ckpt + pb 2. Frozen.pb 3. meta + ??)
答案 0 :(得分:1)
这些文件不是用于存储完整模型的不同文件类型。每个模型都包含不同的信息,并且它们共同存储一个模型。
.meta
文件存储模型的大部分结构,而.ckpt
文件是检查点文件,用于存储有关网络内变量值(神经元的权重和偏差)的信息网络)处于保存网络的训练阶段。 .pb
文件是协议缓冲区文件。
要查看如何使用这些文件保存和还原模型的工作示例,请参见此处:
https://bitbucket.org/FinleyGibson/save-and-load-tf-model/
并使用jupyter笔记本文件查看器打开.ipynb文件