Pytorch ⇛ ONNX
■ Open Neural Network Exchange (ONNX) は共通になろうと
するニューラルネット記述方式
● https://onnx.ai/
■ pytorch_modelにモデルをロード
● 例えば pytorch_model = timm.create_model('mobilenetv2_100')
■ torchに内蔵された関数で変換できる
● torch.onnx.export(model=pytorch_model,
args=torch.ones((1, 3, 224, 224)),
f='./model.onnx',
input_names=['input'],
output_names=['output'])
4