Hey @ilyes. Thanks for the implementation. I was wondering what is y in your forward function? Also, does it mean that the activation has to be time distributed i.e for every conv2d in the ModuleList?