PyTorch モデルのトレーニング

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

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