Hi,my training set is 800,000 rows and 153 columns. The data I want to predict is 200,000 rows and 153 columns. How should I write these 2 lines?
self.fc1 = nn.Linear(800000, 153)
self.fc2 = nn.Linear(153, 2)

If I use this code,the error message is:

RuntimeError:
size mismatch, m1: [800000 x 153], m2: [800000 x 153] at …\aten\src\TH/generic/THTensorMath.cpp:41