ZeRO checkpointing to support torch.optim.Adam #332

Closed

Assignees

opened

on Aug 24, 2020

ZeRO checkpointing currently works only for apex's fused_adam, but not for torch's adam.

Metadata

Assignees

tjruwase

Labels

No labels

No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests