Code

[BE] Lr schduler flatten#794

Code

Merged

mori360 merged 9 commits intopytorch:mainpytorch/torchtitan:mainfrom mori360:lr_schduler_flattenmori360/torchtitan:lr_schduler_flattenCopy head branch name to clipboardJan 31, 2025

+17-12

Commits

Commits on Jan 16, 2025

• flatten lr scheduler

  mori360

  committed
  83d01fa

  Copy full SHA for 83d01fa

• flatten lr scheduler

  mori360

  committed
  dbf1f07

  Copy full SHA for dbf1f07

Commits on Jan 17, 2025

• remove get_lr_scheduler_state

  mori360

  committed
  9da918b

  Copy full SHA for 9da918b

Commits on Jan 29, 2025

• add schedulercontainer as Stateful

  mori360

  committed
  2b7e95f

  Copy full SHA for 2b7e95f

• save state_dict to checkpoint

  mori360

  committed
  c0f061f

  Copy full SHA for c0f061f

Commits on Jan 30, 2025

• save and load only one state_dict for lr_scheduler

  mori360

  committed
  c0a4057

  Copy full SHA for c0a4057

• change for loop

  mori360

  committed
  4d776f0

  Copy full SHA for 4d776f0

• change comment

  mori360

  committed
  12a0bb2

  Copy full SHA for 12a0bb2

Commits on Jan 31, 2025

• update comments

  mori360

  committed
  fd683e6

  Copy full SHA for fd683e6