Thanks a lot! I was able to adjust it.
However, I am still running into an error. I am using a CNN LSTM and I want to return the output for every sequence in the LSTM. When trying to do that, I get the following error:
Expected target size (20, 20), got torch.Size([20, 10])