search

LiJunnan1992 / MLNT

Meta-Learning based Noise-Tolerant Training
123 stars 29 forks source link

how can I run this code in CIFAR data? #13

Open AllenPu opened 3 years ago

AllenPu commented 3 years ago

since the pretrain size is 224 but cifar size is 32, so, directly use this code would cause the out size error, could you please tell me how can I use the cifar 10 data in this code?

AllenPu commented 3 years ago

I tried to use this code https://github.com/LiJunnan1992/DivideMix/blob/master/PreResNet.py but I found that the forward part is different from the MLNT part and if I want to use cifar data, the "fast weight" in main::train can not be forward to the DivideMix part.

yanghao1996 commented 3 years ago

Have you fixed this problem? I also want to run this code on cifar 10.