Hi @CedricLy, thank you for your response!
What do you mean with optimizer.load_state_dict(pathload[])?
But as discussed here, I think you are right. You only need to load the optimizer state if you want to continue training on the same data with the same network. Otherwise the model.state_dict() is sufficient.