Skip to content

Email Callback on training done #20133

Open
@loucaspapalazarou

Description

@loucaspapalazarou

Description & Motivation

A simple callback that accepts a user's email and an app account password to notify the user that their training is complete by email. Inspired by wandb

The call back will use the user's email as both sender and receiver so no need for an external SMTP.

email = johndoe@gmail.com
password = os.getenv("EMAIL_PASSWORD")
email_callback = EmailCallback(email, password)

trainer = Trainer(
        ...
        callbacks=[email_callback],
    )

Screenshot 2024-07-28 124011

Could also add details on how the training went, like test or validation measures. We could also add graphs and make it an entire report. It can get as detailed as we want. It's also possible to notify the used of any potential crashes.

Would love to work on this if approved!

Pitch

A lot of the time, training takes up a number of hours and users keep checking up on their models manually. An automatic notification on a user's phone where they can't miss it will speed up the development process.

Alternatives

Users manually create the callback.

Additional context

No response

cc @Borda

Metadata

Metadata

Assignees

No one assigned

    Labels

    discussionIn a discussion stagefeatureIs an improvement or enhancement

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions