albumentations.torch seems to be renamed as albumentations.pytorch.