Closed shirubei closed 10 months ago
建议改为 DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
Good suggestion, already accepted and updated.
建议改为 DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"