torch-uncertainty icon indicating copy to clipboard operation
torch-uncertainty copied to clipboard
verified publisher icon ENSTA-U2IS-AI

Pretrained parameters

Open ZhouCX117 opened this issue 3 years ago • 4 comments

Hi, thanks for your excellent work and sharing codes. I have a small question. If I apply packed ensemble, the pre-trained parameters of the ResNet on the ImageNet can't be used. Is it right?

ZhouCX117 avatar Mar 10 '23 08:03 ZhouCX117

Hi, thank you for your interest! Indeed, the pretrained parameters of the ResNet on the ImageNet can't be used as such. Therefore, we plan to release as soon as possible our trained ResNet parameters on ImageNet. We will keep you posted on this matter.

alafage avatar Mar 10 '23 13:03 alafage

Hi! Great work! I was wondering, on Hugging Face you have the resnet18 version of packed ensembles trained on CIFAR100, but is there also a version trained on IMAGENET besides the resnet50 version? Bests!

Vujas-Eteph avatar Dec 12 '23 16:12 Vujas-Eteph

Hi @Vujas-Eteph, sorry for the delay.

I've started working once again on the Hugging Face models. In the following days, I'll try to put several checkpoints for Deep Ensembles and the Packed-Ensembles checkpoints on ImageNet.

Happy Holidays!

o-laurent avatar Dec 25 '23 14:12 o-laurent

Hi @o-laurent, no worries about the delay!

It's great hearing that you'll add more checkpoints for the Packed-Ensembles (ImageNet variant) :)

Happy Holidays to you too! ❄️

Vujas-Eteph avatar Dec 28 '23 12:12 Vujas-Eteph