mlfoundations / model-soups

Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference time
MIT License
430 stars 39 forks source link

Hyperparameter config for the CIFAR models #18

Open santosh9sanjeev opened 11 months ago

santosh9sanjeev commented 11 months ago

Hello, Thank you for the great work.

I understand that you provide the finetuned weights for ImageNet as well as the hyperparameter config. Furthermore, you have also privded the models for CIFAR-10.

1) I would kindly request you to also provide the hyperparameter config for the different models finetuned on CIFAR-10 dataset.

2) Also it would be really great if you could also provide the script you trained models on ImageNet or CIFAR datasets, the provided finetune.py has minimal hyperparameters, but I believe you have explored extensive augmentations.

It would be greatly appreciated if you provided the above (atleast 1st one).

Thank you!