导出onnx需要注意的事项

本文根据项目实践总结了9条导出onnx注意事项,我们基本上按照这9条方式去操作,可以减少很多错误,使得模型部署更加顺利。

(1) 任何用到shape、size返回值的参数时,而是加上int转换

例如: tensor.view(tensor.size(0),-1)这类操作,避免直接使用tensor.size的返回值,而是加上int转换 tensor.view(int(tennsor.size(0)),-1),从而断开做值的跟踪。

例如:

import torch
import torch.nn as nn

class Model(nn.Module):
	def __init__

你可能感兴趣的:(object,detection,深度学习,python,pytorch)