type torch.cuda.FloatTensor but found type torch.cuda.ByteTensor
train_label_batch = torch.from_numpy(train_label_batch)
train_label_batch = train_label_batch.type(torch.FloatTensor) # 转Float
train_label_batch = train_label_batch.cuda() # 转cuda
或者train_label_batch.type(torch.cuda.FloatTensor)