It kinda looks like your input tensor might be the wrong size - Conv2d expects a tensor of size (n, k, h, w) where in your case that’s (1, 4, 80, 80) assuming your input is 80x80 and has 4 channels which is what it looks like. I would recommend checking your tensor shape and making sure it’s in the right format