I think you have a size mismatch when passing x into the first linear layer (self.fc1). Please check if x.shape (before passing into fc1) is really 720 in the second dimension.



 Last_One:

x = F.relu(self.fc1(x))