pytorch使用trace模型

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

你可能感兴趣的:(pytorch实践,pytorch)