pytorch使用trace模型1、使用trace生成torchscript模型2、使用trace的模型预测1、使用trace生成torchscript模型def save_trace(model, input, save_path): traced_script_model = torch.jit.trace(model, input) traced_script_model.save(save_path)2、使用trace的模型预测# -*- coding : UTF-8 -*
pytorch实践
2023-10-07
44 点赞
0 评论
66 浏览