I want to create a model with 2*7 input and it has 2 outputs which have been classified from 4096 classes. How can I do it? It is my model. Thank you in advance.
class FeedForwardNNModel(nn.Module): def __init__(self, input_size, hidden_size, num_classes): super(FeedForwardNNModel, self).__init__() # Linear Function self.fc1 = nn.Linear(2 * 7, hidden_dim) # Non-Linearity self.sigmoid = nn.Sigmoid() # Linear function (readout) self.fc2 = nn.Linear(hidden_dim, 4096) def forward(self, x): # Linear function out = self.fc1(x) # Non-linearity out = self.sigmoid(out) # Linear function (readout) out = self.fc2(out) return out