sorry, I couldn’t understand it. I have already cast my input feature and label by train = data_utils.TensorDataset(DEAP_x_train_torch.long(), DEAP_y_train_torch.long（）)     but it’s still  wrong. Did I get correct understanding what you mentioned before?