fattyhan

Results 1 comments of fattyhan

使用mps模式 ``` if torch.backends.mps.is_available(): self.device = torch.device("mps") else: self.device = torch.device("cpu") ```