ArthurRizar/tensorflow_ernie

转为tf2.0代码

16pk opened this issue · 0 comments

16pk commented

你好~tf2.0弃用了session这个概念,你的项目中np_to_tf代码已无法在tf2.0版本中执行。我在网上检索了一番,发现包括官方文档在内的几乎所有资料都是使用checkpoint以模型为单位进行预训练变量的存储及载入的。这里的情景是没有网络结构仅保存预训练参数。请问应该如何实现呢?你有计划更新代码么?