I’m having trouble running your model with CNN_Text(torch.randn(64, 18). In general, though, through some print statements like you’re doing right now you should be able to find out which layer you’re passing in the wrong input to. If it’s one of the conv layers, the docs 34 have a formula for what size the output ends up to be.