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?

[Mahiro2211]

I found that if I deal with the coco dataset with the same method of dealing with nuswide , It works very well
I changed the config

'dataset_kwargs': {
        'resize': 256 if args.ds in ['nuswide' ,'coco'] else 224,
        'crop': 224,
        'norm': 2,
        'evaluation_protocol': 1,  # only affect cifar10
        'reset': False,
        'separate_multiclass': False,
    },
    ```