Skip to content

Improve use_state_dict in AveragedModel #66686

Closed
Closed
Improve use_state_dict in AveragedModel#66686
Assignees
prabhat00155
Labels
module: optimizerRelated to torch.optimtriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
@prabhat00155

Description

@prabhat00155
prabhat00155
opened on Oct 15, 2021

use_state_dict was added in AveragedModel in #65495 and #65921.
This needs to be further improved as described here. This issue has been created to keep track of this work item.

cc @vincentqb @jbschlosser @albanD @datumbox

Metadata

Metadata

Assignees

Labels

module: optimizerRelated to torch.optimtriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions