Hi there,
I would like to add a Softmax layer to the end of a custom Resnet50 that i have trained. I wana add it for validation. My ResNet50 is:
    model = torchvision.models.resnet50(pretrained=True)
    model.fc = nn.Sequential(
        nn.Linear(model.fc.in_features, 1000),
        nn.Linear(1000, 128),
        nn.Linear(128, 64),
        nn.Linear(64, 8)
    )

How could I do it?
Thanks