エントリーの編集
エントリーの編集は全ユーザーに共通の機能です。
必ずガイドラインを一読の上ご利用ください。
PytorchのモデルをTensorflowjsに変換する
記事へのコメント0件
- 注目コメント
- 新着コメント
このエントリーにコメントしてみましょう。
注目コメント算出アルゴリズムの一部にLINEヤフー株式会社の「建設的コメント順位付けモデルAPI」を使用しています
- バナー広告なし
- ミュート機能あり
- ダークモード搭載
関連記事
PytorchのモデルをTensorflowjsに変換する
はじめに 気軽にブラウザで機械学習のモデルを動かしたい、という欲求から、tensorflowjsでいろんなモデ... はじめに 気軽にブラウザで機械学習のモデルを動かしたい、という欲求から、tensorflowjsでいろんなモデルを動かすライブラリを作っています(github)。しかし、下記のグラフに見られるように、ここ最近は、主要カンファレンスの75%のペーパーでPyTorchが使われているようになっており、エッジの効いた面白そうなモデルをtensorflowjsで簡単には動かせない状況になってしまいました。 State of AI Report 2020 (site) ところが、モデル量子化の権威であられるPINTOさんが、今年のAdvent CalenderでPytorchからTensorflow系のモデルへのコンバート手法を公開してくれました(site)。まさに神。ありがたすぎる! 公開された手法を用いてモデルのコンバートを再現してみたところ、私の環境ではいくつか詰まったところがあったので、作業メ