jdhao's blog
·
A Dig into PyTorch Model Loading
📝
内容提要
Saving and loading PyTorch models # Models in PyTorch are a subclass of torch.nn.Module. To save the model parameters, we use model.state_dict() to get all the model parameters:
➡️
