Solved by lelouedec in post #6 
              
              
                What do you want to do exactly, X_train.values is giving you a numpy array, so torch.from_numpy should return correctly a Tensor.