model/transform.py

import torch 
import torchvision.transforms as T  
transforms = T.Compose([
    T.ToTensor(),T.Resize((256,256), antialias=False), T.Lambda(lambda x: x.to(torch.float32))])