Skip to content

Should I save optimizer ? #10

@mudimingquedeyinmoujia

Description

@mudimingquedeyinmoujia

It seems the script main.py and main_dit.py only save model using accelerator.save_model(model, save_dir) and do not save the state of optimizer, so is this a negligence or the character of accelerator ? Will this influence resume training ?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions