thanks  fs4ss1 but  I have another problem
train_x, val_x, train_y, val_y = train_test_split(train_x, train_y, test_size=0.3)
train_x = train_x.reshape(train_x.shape[0], 1, train_x.shape[1], train_x.shape[2])
train_x = torch.from_numpy(train_x)

TypeError: expected np.ndarray (got Tensor)

help me please