Method wrap_dataloader

workers/base.py:148–149

(self)

        self.wrap_dataloader()

    def wrap_dataloader(self):
        self.dataloader = self.accelerator.prepare(self.dataloader)

    def wrap_model(self,):
        self.model = self.accelerator.prepare(self.model)