MCPcopy Create free account
hub / github.com/MarcCoru/locationencoder / train_dataloader

Method train_dataloader

data/inat2018_loader.py:301–302  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

299 self.test_ds = TensorDataset(locations, classes)
300
301 def train_dataloader(self):
302 return DataLoader(self.train_ds, batch_size=self.batch_size, shuffle=True, num_workers=self.num_workers)
303
304 def val_dataloader(self):
305 return DataLoader(self.valid_ds, batch_size=self.batch_size, shuffle=False, num_workers=self.num_workers)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected