Yeah sorry, what I meant is: I both tried
def forward(self, x):
    relu = nn.ReLU()
    sm = nn.Softmax(dim=1)
    for i, FC in enumerate(self.fc_list):
            x = FC(x)
            x = relu(x)
     x = sm(x)
    return x

and
def forward(self, x):
    relu = nn.ReLU()
    sm = nn.Softmax(dim=1)
    x = self.fc_list[0](x)
    x = relu(x)
    x = self.fc_list[1](x)
    x = sm(x)
    return x

neither is working