🚀 Feature
LightningDataModule class를 이용해서 pre-training data pipeline을 개발한다.
Motivation
- single-node에서 pre-training을 위한 data pipeline 개발
Pitch
Additional context
https://pytorch-lightning.readthedocs.io/en/latest/datamodules.html?highlight=LightningDataModule