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

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

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