保存模型:
PATH='E:\\hellopytorch\\venv\\my_model.pt'
torch.save(model.state_dict(), PATH)
加载模型:
import torch
PATH='E:\\hellopytorch\\venv\\my_model.pt'
model= torch.load(PATH)
保存模型:
PATH='E:\\hellopytorch\\venv\\my_model.pt'
torch.save(model.state_dict(), PATH)
加载模型:
import torch
PATH='E:\\hellopytorch\\venv\\my_model.pt'
model= torch.load(PATH)