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模型,下载对应的模型更方便。

0
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程