pytorch 保存模型 (.pt, .pth, .pkl)

.pt, .pth, .pkl 仅仅是后缀不同 一、只保存模型权重参数，不保存模型结构 保存： torch.save(model.state_dict(), 'mymodel.pth') 加载 ...