Giters

spaceml-org / ml4floods

An ecosystem of data, models and code pipelines to tackle flooding with ML

Home Page:https://spaceml-org.github.io/ml4floods/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[DataPrep] PyTorch-Lightning DataModule

jejjohnson opened this issue · comments

commented

We need a PyTorch-Lightning datamodule which will hold all of the methods, e.g. prepare_, download_, train_test_split_, etc.

TODO

  • prepare_data
  • setup
  • train_dataloader
  • val_dataloader
  • test_dataloader

Resources to Learn More