Skip to content

Instantly share code, notes, and snippets.

@aribornstein

aribornstein/DataModuleExample.py

Created February 14, 2021 15:21
Show Gist options
  • Save aribornstein/744a4283cae8ca3bbda1b7b9639807dd to your computer and use it in GitHub Desktop.
Save aribornstein/744a4283cae8ca3bbda1b7b9639807dd to your computer and use it in GitHub Desktop.
Download ZIP
Raw
DataModuleExample.py
class DataModule(pl.LightningDataModule):
def __init__(self, data_dir: str = PATH, batch_size):
# Initalize
def prepare_data(self):
# Download and Preprocess Raw Data to File System or In Memory
def setup(self, stage=None):
# Load, Transform and Split Data
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment