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
169 点赞
2 评论
256 浏览