Method wrap_model

workers/base.py:151–152

(self,)

        self.dataloader = self.accelerator.prepare(self.dataloader)

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

    def unwrap_model(self,): # this is NOT inplace
        return self.accelerator.unwrap_model(self.model)