Skip to content

Instantly share code, notes, and snippets.

@djbyrne

djbyrne/vpg_dataloader.py

Created July 10, 2020 20:30
Show Gist options
  • Save djbyrne/c1bc51c434132ebec6be7c606d46d37b to your computer and use it in GitHub Desktop.
Save djbyrne/c1bc51c434132ebec6be7c606d46d37b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
vpg_dataloader.py
def train_dataloader(self) -> DataLoader:
dataset = ExperienceSourceDataset(self.train_batch)
return DataLoader(dataset=dataset, batch_size=self.batch_size)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment