備忘録. Pytorchで学習したモデルをtorch.onnx.exportで保存後,Visual Studioでmodel Outputが返ってくるまでのお話. 環境: Windows 10 2004 (OSビルド 19041.1052) onnx opset =11 onnx format version = 6 Visual Studio 2019(UWP) CUDA : 10.0 cuDNN : 7.4.1 torch 1.8.1 + cu102 torchvision 0.9.1+cu102 onnxの保存 色々とハマる点がある. ・ModelのInput ・opsetの指定 ・Mo…