模型转换

从pytorch转到onnx（记录踩坑）

主要使用一个库函数torch.onnx.export()。此函数需要的重要参数包括：被转换的模型，模型输入，转换得到的模型存储位置。
各种坑（主要是因为我太菜了><)
这里的input需要和模型需要的输入维度相同。一开始学长给我说是随便1个图片就行，我以为是维度之类的也是随便，后来经过了多次调试才知道是数值随便但是维度不能改变。
转化的模型是学长写的一个基于resnet的模型，有一些改动，和标准的一些格式不一样。所以转的时候需要改一下，首先是模型的输入不能是字典，改成含两个元素的列表形式。最后在调用这个函数的时候将两个输入放在一起列表里面送进去，输出也不能是字典。