MCPcopy Create free account
hub / github.com/catalys1/mae-pytorch / train_dataloader

Method train_dataloader

datamodule.py:69–77  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

67 pass
68
69 def train_dataloader(self):
70 return DataLoader(
71 dataset = self.data_train,
72 batch_size = self.batch_size,
73 num_workers = self.num_workers,
74 pin_memory = self.pin_memory,
75 shuffle = True,
76 drop_last = True
77 )
78
79 def val_dataloader(self):
80 return DataLoader(

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected