I retained a densenet121 model for binary classification. please How do I save and reload this model to fit with the architecture and the saved parameters.
Thanks in advance
You should you the model’s state_dict to save and load it again. I’d recommend you to take a look at this tutorial.