请问怎么在测试时一次性加载好模型参数并初始化，防止后面在测试时每次都加载模型
1个回答
保存的代码：
torch.save(model.state_dict(), PATH)
加载的代码：
model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
SofaSofa数据科学社区DS面试题库 DS面经
飞时过
2021-09-22 13:55
相关讨论
随便看看