# during traning/val phase make a list of transforms to be used. # input-->"phase",mean,std # output-->list def get_transform(phase,mean,std): list_trans=[] if phase=='train': list_trans.extend([HorizontalFlip(p=0.5)]) list_trans.extend([Normalize(mean=mean,std=std, p=1), ToTensor()]) #normalizing the data & then converting to tensors list_trans=Compose(list_trans) return list_trans