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

Method test_dataloader

datamodule.py:255–263  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

253 self.data_val, self.data_test = random_split(data_val_test, [val_size, test_size])
254
255 def test_dataloader(self):
256 return DataLoader(
257 dataset = self.data_test,
258 batch_size = self.batch_size,
259 num_workers = self.num_workers,
260 pin_memory = self.pin_memory,
261 shuffle = False,
262 drop_last = True
263 )
264

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected