The Conv2D op currently only supports the NHWC tensor format on the CPU. The op was given the format: NCHW

YoujiaZhang / AlphaGo-Zero-Gobang

Meta-Zeta是一个基于强化学习的五子棋(Gobang)模型，主要用以了解AlphaGo Zero的运行原理的Demo，即神经网络是如何指导MCTS做出决策的，以及如何自我对弈学习。源码+教程

MIT License

76 stars 8 forks source link

The Conv2D op currently only supports the NHWC tensor format on the CPU. The op was given the format: NCHW #3

Open jswtyc opened 2 years ago

jswtyc commented 2 years ago

您好我在克隆你的代码时，发现如上报错无法运行。（一番定位后应该是PolicyNN.py文件中的 act_probs, value = self.model.predict_on_batch(currentBoard) 这句话报错，但是无奈本人水平有限，网上资料甚少，无法解决。）个人怀疑是环境问题，多次重装readme文件中提及的tensorflow-gpu均无果。故请问是否方便透露下此代码的具体环境，以便克隆复现。

KarsonYu commented 1 year ago

我也是，请问你解决了吗

jswtyc commented 1 year ago

解决了，具体怎么回事忘了，反正是环境的原因。作者代码写的还是不错的。

elysia-best commented 10 months ago

我也是，请问你解决了吗

数据格式的问题，切换到正确安装tensorflow cuda的环境下就可以正常处理了