下面代码的功能是先训练一个简单的模型,然后保存模型,同时保存到一个pb文件当中,后续可以从pd文件里读取权重值。

tensorflow 保存模型和取出中间权重例子

tensorflow 保存模型和取出中间权重例子