Hello Pytorchers,
I am having an issue with trained model. In training, model after time performed very good, but after it was saved and loaded (state_dicts), it could not even give 1 valid prediction.
Network structure:
    def __init__(self):
        super(Net, self).__init__()
        self.fc1 = nn.Linear(64, 128)
        self.sig = nn.Sigmoid() # Tried also Tanh and ReLu
        self.fc2 = nn.Linear(128, 64)

Optimizer: RMSprop
Criterion: SmoothL1Loss
P.s. Model size is like 66kb, is it okay?
P.s. Tried different hidden neuron size ( 32, 64, 128 ). Tried different learning rates.
Thank you!