model import torch.nn as nn class Classifier(nn.Module): def __init__(self): super(Classifier, self).__init__() self.fc1 = nn.Linear(in_features=2, out_features=16) self.fc_act1 = nn.ReLU() self.fc2 = nn.Linear(in_features=16, out_features=32) self.fc_act2 = nn.ReLU() self.fc3 = nn.Linear(in_features=32, out_features=4) def forward(self, x): x = self.fc1(x) x = self.fc_act1(x) x = self.fc2(x) x ..