Skip to content

Support teardown for Lightning DataModule #4418

@ananthsub

Description

@ananthsub

🚀 Feature

teardown as a hook can be useful for data modules.

Motivation

This could be used for:

  • Clean up downloaded data after training finishes
  • Closing any open connections a dataloader makes
  • etc

Pitch

This has natural connections to prepare_data and setup and could be implemented very similarly to how those are supported across the data module and lightning module.

By default this should do nothing

cc @nateraw

Metadata

Metadata

Assignees

No one assigned

    Labels

    data handlingGeneric data-related topicfeatureIs an improvement or enhancementhelp wantedOpen to be worked onwon't fixThis will not be worked on

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions