I don’t know which code path is taken in your __getitem__ method, but since ToTensorV2 uses torch.from_numpy directly (and numpy uses float64 by default) I would assume this code path keeps the dtype as float64.