class ExNet(nn.Module): def __init__(self): super(ExNet,self).__init__() self.fc1 = nn.Linear(1024,500) self.fc2 = nn.Linear(500,100)