AI学习

ONNX

• 转pth模型到ONNX一定要遵循3个原则：
  • 在调用onnx.export 的API前， 一定要时model.eval()
  • 在调用onnx.export 的API前， 一定时刚刚load_state_dict, 之后模型没有任何修改
  • 在调用onnx.export的API前， 模型在正常推理的时候所有处理，转之前都要做