since  i’m taking the .values of a dataframe its using the numpy array. and in def __getitem__(self,idx) if i return only return torch.tensor(target[idx]) it works fine. and also i have printed the shape of target it’s tensor([2000])