import random import torch ... self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") ... return random.choice(self.class_0_samples)
確定! 回上一頁