alshedivat / lola

Code release for Learning with Opponent-Learning Awareness and variations.
MIT License
145 stars 35 forks source link

Transpose of input #2

Open a7b23 opened 6 years ago

a7b23 commented 6 years ago

In coin game the input shape is [4,grid_size,grid_size]. Shouldn't the input be transposed before passing to the model in networks.py so that the channel(4) becomes the last dimension?