PyTorch モデルのトレーニング
PyTorch プロジェクトは、ディープ ラーニング ネットワークを構築するため、GPU アクセラレータ テンザー評価と高レベルの機能を提供する Python パッケージです。
MLflow PyTorch ノートブックは、MNIST の手書き数字認識データ上のニューラル ネットワークに適合し、ログの実行結果を MLflow サーバーに送信します。 トレーニング メトリックと重みを TensorFlow イベント形式でローカルにログに記録し、MLflow 実行の成果物ディレクトリにアップロードします。 最後に、TensorBoard を起動し、ローカルでログに記録されたイベントを読み取ります。
準備ができたら、Azure Databricks で提供されるモデルを使用してモデルをデプロイできます。