minor fix in tensorflow support

junxiaosong / AlphaZero_Gomoku

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

MIT License

3.27k stars 964 forks source link

Closed dshnightmare closed 6 years ago

dshnightmare commented 6 years ago

Add data_format parameter in cnn layers for channels_first data