PyTorch モデルをトレーニングする

PyTorch は、GPU アクセラレーションによるテンソル計算と、ディープラーニング ネットワークを構築するための高レベルの機能を提供する Python パッケージです。

MLflow PyTorch ノートブックは、MNIST手書き数字認識データ上のニューラルネットワークに適合し、実行結果をMLflowサーバーに記録します。 トレーニングメトリクスと重みを TensorFlow イベント形式でローカルにログに記録し、MLflow 実行のアーティファクトディレクトリにアップロードします。 最後に、TensorBoard を起動し、ローカルに記録されたイベントを読み取ります。

準備ができたら、 Databricksのモデルサービングを使用してモデルをデプロイできます。

MLflowのPyTorch モデルトレーニングノートブック

ノートブックを新しいタブで開く