Hi the code is just for a cell, I updated the gist with how the sequence loop could be used.
About the requires_grad, I use the nn.Conv2d module, isn’t that automatic when using nn modules?
Could anybody confirm?