Thank you for you reply. I found out that when I dropped .cuda(), self.filter1 and self.filter2 appear in model.parameters(), that’s where I got those extra shape. Further in this case, self.bias1 and self.bias2 have value nan, although self.theta1, self.Lambda1, self.psi1 have their value. This makes me quite confused.