MCPcopy Create free account
hub / github.com/Text-to-Audio/Make-An-Audio / _predict_dataloader

Method _predict_dataloader

main.py:235–238  ·  view source on GitHub ↗
(self, shuffle=False)

Source from the content-addressed store, hash-verified

233 num_workers=self.num_workers, worker_init_fn=init_fn, shuffle=shuffle)
234
235 def _predict_dataloader(self, shuffle=False):
236 init_fn = None
237 return DataLoader(self.datasets["predict"], batch_size=self.batch_size,
238 num_workers=self.num_workers, worker_init_fn=init_fn)
239
240
241class SpectrogramDataModuleFromConfig(DataModuleFromConfig):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected