If you just pass the tmp_dset_train.data and tmp_dset_train.targets to your custom Dataset, you are missing the transformations (i.e. ToTensor() in this case), so that your data won’t be normalized.
Add the transform to your custom Dataset and try to run it again.