DeepCoral Algorithm in Pytorch

Question

سلام. من کد الگوریتم DeepCoral رو میخوام برروی داده MNIST که 28*28 هست اجرا کنم. در تابع data_loader مربوط به کد قسمت ای مه در پایین کدش هست رو تغییر میدم به 28 در 28 به جای 256 در 256، اما کد خطا داره. لطفا راهنمایی بفرمایید. لینک کد را در ادامه گذاشتم. با تشکر فراوان

https://github.com/jindongwang/transferlearning-tutorial

transform = {
        'train': transforms.Compose(
            [transforms.Resize([256, 256]),
                transforms.RandomCrop(224),
                transforms.RandomHorizontalFlip(),
                transforms.ToTensor()]),
        'test': transforms.Compose(
            [transforms.Resize([224, 224]),
                transforms.ToTensor()])
    }