search

mlech26l / ncps

PyTorch and TensorFlow implementation of NCP, LTC, and CfC wired neural models
https://www.nature.com/articles/s42256-020-00237-3
Apache License 2.0
1.94k stars 311 forks source link

Example of stacking LTC with convolutional layers on pytorch version #51

Closed wyl121 closed 1 year ago

wyl121 commented 1 year ago

Hello!Thank you for having such a great work for many beginners like me to learn, recently I am using pytorch framework to build LTC model, but I can never use convolutional layer and LTC well together, always report the error of data dimension mismatch makes me very distressed, hope to get a pytorch framework of convolutional layer and LTC stacking example, looking forward to your reply!

mlech26l commented 1 year ago

Hello, The Atari behavior cloning tutorial contains an example of convolutional layers stacked with an LTC/CfC: https://ncps.readthedocs.io/en/latest/examples/atari_bc.html

wyl121 commented 1 year ago

Thank you very much for your help, I have benefited a lot!