Trainer for SimCLR network?

Jingkang50 / OpenOOD

Benchmarking Generalized Out-of-Distribution Detection

MIT License

858 stars 108 forks source link

Trainer for SimCLR network? #155

Closed 2454511550Lin closed 1 year ago

2454511550Lin commented 1 year ago

Hi, I tried to train Resnet-50 with SimCLR loss on Cifar10 from scratch. The scrip is as follows: python main.py --config configs/datasets/cifar10/cifar10.yml configs/preprocessors/base_preprocessor.yml configs/networks/simclr.yml configs/pipelines/train/?.yml I am wondering which trainer from openood/trainers/ should I choose?

Besides, I am wondering if there are pre-trained SimCLR ResNet models for Cifar-10/100 available. Thank you so much.

zjysteven commented 1 year ago

I don't think we provide the exact SimCLR trainer. However, CIDER and NPOS uses SimCLR-like contrastive training if I remember correctly.