tf1保存的模型怎么迁移到tf2

Question

老师，tf1保存的ckpt tf2怎么读取，我把bert的模型下载下来了，但是不会转化成model，因为里面的格式是ckpt的

Answer 1

正十七 2021-07-23 07:39:22

同学你好，ckpt的load可以参考：https://www.tensorflow.org/api_docs/python/tf/train/Checkpoint

但是需要你拿到原来的bert模型的模型结构才可以，尤其是要用tf.keras实现的话会跟方便。

另外，如果你只是想用bert，建议用 huggingface的transformer库 https://github.com/huggingface/transformers，里面直接用tf2实现的bert模型，下载对应的模型更方便。