search

hendrycks / ss-ood

Self-Supervised Learning for OOD Detection (NeurIPS 2019)
MIT License
266 stars 31 forks source link

how can I reproduce your imageNet AUROC? #17

Closed sh524shin closed 3 years ago

sh524shin commented 3 years ago

First and Foremost , I really appreciate you for sharing code. Now I try to reproduce ImageNet AUROC 85.7%, but I don't know exact parameter setting(eg learning rate, epoch etc..) and also ImageNet dataset downloaded from your github is enough for training? plz let me know~ thanks!

hendrycks commented 3 years ago

Download the one class ImageNet test set here. The one class ImageNet training set is here.

The training set should be enough for training. I think https://github.com/hendrycks/ss-ood/blob/master/train.py should have reasonable hyperparameters. The code does not have resize prediction, but that only slightly helped performance.