How to use ml flow with pytorch?

[Dawid]

Excited to learn how can I efficiently and quickly log metrics in ml flow from my pytorch model? #summit23

[Kumaran · Accepted solution]

Hello @Dawid 
Please check the below notebook that is 'MNIST handwritten digit recognition data' with mlflow with pytorch.

https://docs.databricks.com/_extras/notebooks/source/mlflow/mlflow-pytorch-training.html

https://docs.databricks.com/mlflow/tracking-ex-pytorch.html

[Anonymous]

Hi @Dawid 

Hope you are well. Just wanted to see if you were able to find an answer to your question and would you like to mark an answer as best? It would be really helpful for the other members too.

Cheers!