请问如何用tensorflow加载模型文件

junxiaosong / AlphaZero_Gomoku

An implementation of the AlphaZero algorithm for Gomoku (also called Gobang or Five in a Row)

MIT License

3.25k stars 965 forks source link

请问如何用tensorflow加载模型文件 #64

Open moonsin opened 5 years ago

moonsin commented 5 years ago

请问给的best_policy_8_8_5.model 怎么用tensorflow加载啊。直接用这个路径创建PolicyValueNet类会报错。谢谢！

junxiaosong commented 5 years ago

给的模型文件是最开始基于Theano/Lasagne的版本训练得到的，所以没法直接用tensorflow加载。但是如果你用cPickle加载那个文件的话可以看到其实保存的就是神经网络每一层的权重参数的list，你可以考虑如何将这些权重参数赋值给tensorflow的网络，之前在PyTorch里类似处理过，参见 #5