Downloads · 30 days
0
erow/rn50_simclr_bs4096_e1000
rn50_simclr_bs4096_e1000 is a machine learning model from erow. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch. The card lists the license as apache-2.0.
ResNet-50 SimCLR pretrained on ImageNet-1K with FastSSL (mainpretrain.py, configs/simclrif.gin).
Downloads · 30 days
0
Access
Public
Updated Sep 19, 2026
Repo size
4.8 GB
Likes
0
Public
Click a slice to open those files.
.pth4.8 GB · 100%
From the Hugging Face model README
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.