Method print

byol_pytorch/trainer.py:111–112

(self, msg)

        return self.accelerator.wait_for_everyone()

    def print(self, msg):
        return self.accelerator.print(msg)

    def forward(self):
        step = self.step.item()