AozhongZhang / COMQ

6 stars 0 forks source link

Is there another way to get the pretrained model? #6

Open qifeng22 opened 4 months ago

qifeng22 commented 4 months ago

image I tried many times, the original model test accuracy is bad. Can you upload the pretrained model? image I replace my model file.

Or, Can you help analyze the reason that caused it? @AozhongZhang

AozhongZhang commented 4 months ago

@qifeng22 Maybe the data transforms didn't match your model. How did you set the data transforms config when you trained your model? Probably you should use the configuration that you trained with.

qifeng22 commented 4 months ago

image Your code don't train the model, the model pretrained file is from https://huggingface.co/timm/vit_small_patch16_224.augreg_in21k_ft_in1k, the train_transforms and test_transforms as show from the figure. Did you say data transforms denote it? @AozhongZhang

AozhongZhang commented 4 months ago

@qifeng22 yes, this is what I said data transforms.

qifeng22 commented 4 months ago

ok, I use the transforms config from pretrained file. And don't konw why original model accuracy so low in test dataset. Can you upload the pretrained file. @AozhongZhang

AozhongZhang commented 4 months ago

@qifeng22 OK, I will upload soon.

qifeng22 commented 4 months ago

okok, thanks