KAI-YUE / rog

10 stars 5 forks source link

CIFAR10 #2

Closed HuitongJin closed 4 months ago

HuitongJin commented 6 months ago

Hello, if I want to use this algorithm on the CIFAR10 dataset, do the pre-training weights in model_zoos need to be adjusted? How to train the weights for CIFAR10? The files involved include: postmodel.pth, tresnet.pth, usrgan.pth etc.

KAI-YUE commented 4 months ago

I am not sure which resolution you want to preserve (128 or 32). You can align the labels of cifar-10 to Imagenet and give a try first.