冬は寒いのでDNNの学習を回すのにぴったり！GPUの廃熱で暖房費節約だぜ！などと思ったけれど、メインマシンはメインOSはWindowsで運用していてビデオカードもAMDなのでDNNフレームワークを動かすのはしんどい。 調べたらDirectMLってのでWindows+AMDでもいけ…

やりたかったこと PyTorchモデルをONNXファイルにexportしたい モデルにはカスタムオペレータを含む カスタムオペレータの実装は存在しない（該当部分は推論不能） カスタムオペレータはONNXでそれらしきop_typeを持つノードで出力してくれればOK やってみた…

前回の続き？のような何か。 maminus.hatenadiary.org 今回やったこと ROCm3.5 Docker環境 PyTorchをmasterブランチでビルド torchvisionをmasterブランチでビルド 上記に伴い公式にバグ報告 前回同様PyTorch版EfficientDet1で動作確認 前回からの変化点を中…

PyTorch版EfficientDet1のカメラで撮影した画像に推論結果を重畳してくれるデモをROCm環境で動かしたので手順などをメモ。 環境 Radeon RX 580 Ubuntu 18.04.4 LTS（ホスト側） dockerイメージ Docker Hubでrocm/pytorchの中からそれっぽいのを選択。 Ubuntu…