方法一:
targets =torch.LongTensor(targets)
方法二:
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
targets = targets.to(device,dtype=torch.int64)
方法一:
targets =torch.LongTensor(targets)
方法二:
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
targets = targets.to(device,dtype=torch.int64)