MCPcopy Create free account
hub / github.com/CompVis/diff2flow / _train_dataloader

Method _train_dataloader

diff2flow/dataloader.py:242–244  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

240 self.test_dataloader = self._test_dataloader
241
242 def _train_dataloader(self):
243 return DataLoader(self.datasets["train"], batch_size=self.batch_size,
244 num_workers=self.num_workers, shuffle=True)
245
246 def _val_dataloader(self):
247 return DataLoader(self.datasets["validation"], batch_size=self.val_batch_size,

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected