jdhao's digital space

Recent content on jdhao's digital space

马上订阅 jdhao's digital space RSS 更新: https://jdhao.github.io/index.xml

A Dig into PyTorch Model Loading

2022年1月28日 23:17

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:

查看完整文章