Hello,I have some question about the dataset COCO

[Mahiro2211]

Thanks for sharing the code, but When I followed the instruction and got ready to test it on COCO I met some error

RuntimeError: stack expects each tensor to be equal size, but got [3, 297, 224] at entry 0 and [3, 224, 335] at entry 1

At first, I think the question was about the transform

'coco': [
                    transforms.Resize(resize),
                    transforms.RandomCrop(crop),
                    transforms.RandomHorizontalFlip()
                ]

but I tried several times to change the transform, still wrong, would you please do me a favor?