このセクションには、 tensorflow
パッケージとは別のプロジェクトとパッケージのAPIリファレンスページの追加コレクションが含まれていますが、専用のサブサイトページはありません。
TensorFlowモデルリポジトリ
TensorFlowモデルリポジトリは、最先端の(SOTA)モデルの実装を提供します。
official / projectsディレクトリには、TensorFlowの高レベルAPIを使用するSOTAモデルのコレクションが含まれています。これらは、最新のTensorFlow APIを使用して、適切に維持、テスト、および最新の状態に保つことを目的としています。
これらのモデルの構築とトレーニングに使用されるライブラリコードは、pipパッケージとして利用できます。次を使用してインストールできます。
$ pip install tensorflow-models-official # For the latest release
$ #or
$ pip install tf-models-nightly # For the nightly build
ソースからパッケージをインストールするには、次の手順を参照してください。
tensorflow-models-official
pipパッケージには、 tensorflow_models
とorbit
の2つのトップレベルモジュールが含まれています。次のコマンドでインポートできます。
import tensorflow_models as tfm
import orbit
Tensorflowモデル
tensorflow_models
モジュールは、モデルの構築とトレーニングの構成を処理します。アプリケーション固有の機能は、 tfm.vision
およびtfm.nlp
で使用できます。
軌道
orbit
モジュールは、TensorFlowでカスタマイズされたトレーニングループコードを作成するための柔軟で軽量なライブラリを定義します。 Orbitは、動作するモデルのタイプについて柔軟です。 Orbitを使用してKerasモデルをトレーニングできます(KerasのModel.fit
の代わりに)が、Kerasを使用する必要はまったくありません。 Orbitはtf.distribute
とシームレスに統合され、さまざまなデバイスタイプ(CPU、GPU、およびTPU)での実行をサポートします。