GTX_AI 发表于 2020-12-25 21:27:36

pytorch模型保存和加载


.pt模型加载
    net = torch.jit.load(model_path).to(device)
.pth模型加载
    net.load_state_dict(torch.load(model_path))


页: [1]
查看完整版本: pytorch模型保存和加载