cancel
Showing results forย 
Search instead forย 
Did you mean:ย 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
cancel
Showing results forย 
Search instead forย 
Did you mean:ย 

How to use ml flow with pytorch?

Dawid
New Contributor II

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

1 ACCEPTED SOLUTION

Accepted Solutions

Kumaran
Databricks Employee
Databricks Employee
2 REPLIES 2

Kumaran
Databricks Employee
Databricks Employee

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
Not applicable

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!

Connect with Databricks Users in Your Area

Join a Regional User Group to connect with local Databricks users. Events will be happening in your city, and you wonโ€™t want to miss the chance to attend and share knowledge.

If there isnโ€™t a group near you, start one and help create a community that brings people together.

Request a New Group