How to creat a classification model with multiple input and 2 output?

1

Hi,
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