tf1保存的模型怎么迁移到tf2
来源:1-4 Tensorflow2.0架构
小神经中的二百五
2021-07-21
老师,tf1保存的ckpt tf2怎么读取,我把bert的模型下载下来了,但是不会转化成model,因为里面的格式是ckpt的
写回答
1回答
-
正十七
2021-07-23
同学你好,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模型,下载对应的模型更方便。
00
相似问题