my batch size is just 1 .
image tensor size is also less than = 3,1024,1024 .
I dont get such issue with this model which i use for inference  through Dataloader.Since data loader has constraint to keep same size of all image in batch so i thought of building this pipeline