Hugging Face's logo

DM-Diaz
/
SimCLR-ResNet18-ImageNet100

PyTorch
simclr-resnet18
computer-vision
self-supervised-learning
simclr
resnet18
imagenet
lightly
visual-neuroscience
neural-encoding
Model card Files
xet
 Community
SimCLR-ResNet18-ImageNet100
File size: 1,830 Bytes
42018d8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
{
  "model_type": "simclr-resnet18",
  "framework": {
    "training": "Lightly",
    "backend": "PyTorch Lightning"
  },
  "architecture": {
    "backbone": "resnet18",
    "backbone_feature_dim": 512,
    "projection_head": {
      "implementation": "lightly.models.modules.heads.SimCLRProjectionHead",
      "input_dim": 512,
      "hidden_dim": 512,
      "output_dim": 128
    }
  },
  "objective": {
    "loss": "NTXentLoss",
    "temperature": 0.1
  },
  "optimizer": {
    "name": "LARS",
    "momentum": 0.9,
    "weight_decay": 1e-06
  },
  "scheduler": {
    "name": "CosineWarmupScheduler",
    "interval": "step",
    "warmup_epochs": 10
  },
  "input": {
    "size": [
      224,
      224
    ],
    "channels": 3
  },
  "paper": {
    "title": "Eccentricity-Constrained CNN Training Reveals Adaptive Information Coding Around the Visual Field",
    "authors": [
      "Dylan M. Diaz",
      "Margaret M. Henderson"
    ],
    "year": 2026,
    "doi": "10.32470/0416gfsq",
    "arxiv": "2607.19316"
  },
  "notes": [
    "This configuration describes the reference-model training setup used in the associated study.",
    "The exact installed Lightly and PyTorch Lightning package versions were not stored in this configuration."
  ],
  "repository": "DM-Diaz/SimCLR-ResNet18-ImageNet100",
  "checkpoint": "checkpoint_120-resnet18-simclr-imagenet100.ckpt",
  "training": {
    "dataset": {
      "huggingface_id": "clane9/imagenet-100",
      "num_classes": 100,
      "local_format": "ImageFolder-style train/val directories"
    },
    "epochs": 120,
    "batch_size": 64,
    "base_learning_rate": 0.3,
    "lr_scaling_rule": "base_learning_rate * batch_size / 256",
    "initial_learning_rate": 0.075,
    "distributed": false,
    "mixed_precision": "16-mixed",
    "knn_k": 20,
    "knn_temperature": 0.1
  }
}