Method unwrap_model

workers/base.py:154–155

(self,)

        self.model = self.accelerator.prepare(self.model)

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