tensorflow2.0如何用callback回调函数保存pb文件

之前保存ckpt文件只有两文件，一个是index，还有一个data，没有meta文件。看了别的博主分享，需要三个文件才能去转化，所以现在无法转化成pb文件。

目前想要通过callback回调函数去保存pb文件？请问如何实现，能否直接保存为.pb文件，还是需要有固定的代码才可以生成pb文件？

callbacks = [tf.keras.callbacks.ModelCheckpoint(filepath='./save_weights/mymodel_1.pb',
                                                save_best_only=True,
                                                save_weights_only=False,
                                                monitor='val_loss')]

请问可以吗？如果不可以的话，大佬们可以帮忙解决一下吗？本人小白，最近毕设需要将模型保存为pb文件，然后移植到android....T0T谢谢大佬们！

 

不知道你这个问题是否已经解决, 如果还没有解决的话:

• 你看下这篇博客吧, 应该有用👉 ：tensorflow2.0学习笔记：回调函数(callbacks)

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 以帮助更多的人 ^-^