在Torch中,您可以使用torch.save()函數來保存模型,并使用torch.load()函數來加載模型。
保存模型示例代碼:
# 保存模型
torch.save(model.state_dict(), 'model.pth')
加載模型示例代碼:
# 加載模型
model = Model()
model.load_state_dict(torch.load('model.pth'))
model.eval()
在加載模型之后,記得調用model.eval()來將模型設置為評估模式。