Code

Implements MLFlowLogger#2365

Code

Merged

joecummings merged 8 commits intopytorch:mainpytorch/torchtune:mainfrom nathan-az:mlflow-loggernathan-az/torchtune:mlflow-loggerCopy head branch name to clipboardFeb 12, 2025

+200

Commits

Commits on Feb 8, 2025

• Implements MLFlowLogger and basic testing suite

  Show description for 39fc7c9
  nathan-az

  committed
  39fc7c9

  Copy full SHA for 39fc7c9

• Remove duplicated interface

  Show description for 6df0022
  nathan-az

  committed
  6df0022

  Copy full SHA for 6df0022

Commits on Feb 11, 2025

• Add mlflow to dev dependencies

  Show description for cc54013
  nathan-az

  committed
  cc54013

  Copy full SHA for cc54013

• Remove unnecessary None default from getenv

  Show description for 6d4f859
  nathan-az

  committed
  6d4f859

  Copy full SHA for 6d4f859

• Fix logger docstring - delete removed param, add mlflow website

  Show description for 162bc3c
  nathan-az

  committed
  162bc3c

  Copy full SHA for 162bc3c

• Update flatten_dict with reordered args and docstring

  Show description for 675eadf
  nathan-az

  committed
  675eadf

  Copy full SHA for 675eadf

• if x -> if x is not None

  Show description for 7531700
  

  nathan-az

  and

  joecummings

  authored
  7531700

  Copy full SHA for 7531700

• Force kwarg specification in flatten_dict and fix docstring

  Show description for c069438
  nathan-az

  committed
  c069438

  Copy full SHA for c069438