Pythonのための機械学習ライブラリPyTorchの開発チームは、PyTorchと連携可能なコンピュータビジョン用ライブラリ「TorchVision」に、マルチウェイトをサポートするモデルを構築するためのAPIのプロトタイプを12月22日(現地時間)に公開した。
新たなAPIのプロトタイプは、同じモデルバリアントに事前にトレーニングされたさまざまな重みをロードでき、分類ラベルなどの重要なメタデータを追跡し、モデルの使用に必要な前処理変換が可能となっている。
同APIは現在、完成に向けてユーザーからのフィードバックを求めている段階であり、TorchVisionのNightlyバージョンでのみ利用できる。
- 関連リンク
この記事は参考になりましたか?
- この記事の著者
-
CodeZine編集部(コードジンヘンシュウブ)
CodeZineは、株式会社翔泳社が運営するソフトウェア開発者向けのWebメディアです。「デベロッパーの成長と課題解決に貢献するメディア」をコンセプトに、現場で役立つ最新情報を日々お届けします。
※プロフィールは、執筆時点、または直近の記事の寄稿時点での内容です