rn50_simclr_bs4096_e1000

ResNet-50 SimCLR pretrained on ImageNet-1K with FastSSL (main_pretrain.py, configs/simclr_if.gin).

Method SimCLR (build_model.model_fn=@SimCLR)
Backbone ResNet-50, projector 2048-2048-256, τ=0.1
Data ImageNet-1K, 224², two global crops
Optim LARS, lr=0.2, wd=1e-4, warmup 40, 1000 epochs planned
Batch 4×1024 = 4096
Latest checkpoint.pth (epoch 965 / 1000)

Periodic snapshots every 50 epochs: checkpoint-0.pth … checkpoint-950.pth. Training configs are config.yml and config.gin.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support