Rookie

I also got the same error. when my network structure is: ```python model = nn.Sequential( nn.BatchNorm1d(self.feater_dim, affine=self.is_train), nn.Linear(in_features=self.feater_dim, out_features=self.config['hidden1']), nn.GELU(), nn.Linear(in_features=self.config['hidden1'], out_features=self.config['hidden2']), nn.GELU(), nn.Linear(in_features=self.config['hidden2'], out_features=self.config['hidden3']), nn.GELU(), nn.Linear(in_features=self.config['hidden3'], out_features=self.action_dim), nn.Softmax(dim=1) )...