Skip to content

checkpoint will be saved twice at the end of training when save_strategy is epoch #36203

@AaronZLT

Description

@AaronZLT

Reproduction

This is caused by

control.should_save = True

and

control.should_save = True

Saving will happen at the end of training and the end of epoch (which is also triggered when setting save_strategy to epoch). Same model will be saved twice, also, the on_save function will be triggered twice.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions