It seems you’ve now changed the in_features of the linear layer in self.long to 2592100, which is also wrong.
If your input has a shape of [batch_size, 2], use nn.Linear(2, hid) for the first linear layer in self.short and self.long.