@promach the size mismatch error is due to incorrect sizes between layers. So you might have messed up with the Net sizes somewhere in the code. The easiest way to get the right size is to add a print statement in your forward pass and just look at the size. I would suggest looking at you net again.