Sdílet prostřednictvím


Trénování modelu PyTorch

PyTorch je balíček Pythonu, který poskytuje výpočty tensoru akcelerované gpu a vysokou úroveň funkcí pro vytváření sítí hlubokého učení.

Poznámkový blok MLflow PyTorch odpovídá neurální síti na ručně psaných datech rozpoznávání číslic MNIST a protokoly spouštějí výsledky na server MLflow. Protokoluje trénovací metriky a váhy ve formátu událostí TensorFlow místně a pak je nahraje do adresáře artefaktů spuštění MLflow. Nakonec spustí TensorBoard a přečte události zaznamenané místně.

Až budete připravení, můžete model nasadit pomocí modelu obsluhujícího azure Databricks.

Trénovací poznámkový blok modelu MLflow PyTorch

Získat poznámkový blok