73 lines
2.7 KiB
YAML
73 lines
2.7 KiB
YAML
Collections:
|
|
- Name: SimCLR
|
|
Metadata:
|
|
Training Data: ImageNet-1k
|
|
Training Techniques:
|
|
- LARS
|
|
Training Resources: 8x V100 GPUs (b256), 16x A100-80G GPUs (b4096)
|
|
Architecture:
|
|
- ResNet
|
|
- SimCLR
|
|
Paper:
|
|
Title: A simple framework for contrastive learning of visual representations
|
|
URL: https://arxiv.org/abs/2002.05709
|
|
README: configs/simclr/README.md
|
|
|
|
Models:
|
|
- Name: simclr_resnet50_16xb256-coslr-200e_in1k
|
|
Metadata:
|
|
Epochs: 200
|
|
Batch Size: 4096
|
|
FLOPs: 4109364224
|
|
Parameters: 27968832
|
|
Training Data: ImageNet-1k
|
|
In Collection: SimCLR
|
|
Results: null
|
|
Weights: https://download.openmmlab.com/mmselfsup/1.x/simclr/simclr_resnet50_16xb256-coslr-200e_in1k/simclr_resnet50_16xb256-coslr-200e_in1k_20220825-4d9cce50.pth
|
|
Config: configs/simclr/simclr_resnet50_16xb256-coslr-200e_in1k.py
|
|
Downstream:
|
|
- resnet50_simclr-200e-pre_8xb512-linear-coslr-90e_in1k
|
|
- Name: simclr_resnet50_16xb256-coslr-800e_in1k
|
|
Metadata:
|
|
Epochs: 200
|
|
Batch Size: 4096
|
|
FLOPs: 4109364224
|
|
Parameters: 27968832
|
|
Training Data: ImageNet-1k
|
|
In Collection: SimCLR
|
|
Results: null
|
|
Weights: https://download.openmmlab.com/mmselfsup/1.x/simclr/simclr_resnet50_16xb256-coslr-800e_in1k/simclr_resnet50_16xb256-coslr-800e_in1k_20220825-85fcc4de.pth
|
|
Config: configs/simclr/simclr_resnet50_16xb256-coslr-800e_in1k.py
|
|
Downstream:
|
|
- resnet50_simclr-800e-pre_8xb512-linear-coslr-90e_in1k
|
|
- Name: resnet50_simclr-200e-pre_8xb512-linear-coslr-90e_in1k
|
|
Metadata:
|
|
Epochs: 90
|
|
Batch Size: 4096
|
|
FLOPs: 4109464576
|
|
Parameters: 25557032
|
|
Training Data: ImageNet-1k
|
|
In Collection: SimCLR
|
|
Results:
|
|
- Task: Image Classification
|
|
Dataset: ImageNet-1k
|
|
Metrics:
|
|
Top 1 Accuracy: 66.9
|
|
Weights: https://download.openmmlab.com/mmselfsup/1.x/simclr/simclr_resnet50_16xb256-coslr-200e_in1k/resnet50_linear-8xb512-coslr-90e_in1k/resnet50_linear-8xb512-coslr-90e_in1k_20220825-f12c0457.pth
|
|
Config: configs/simclr/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py
|
|
- Name: resnet50_simclr-800e-pre_8xb512-linear-coslr-90e_in1k
|
|
Metadata:
|
|
Epochs: 90
|
|
Batch Size: 4096
|
|
FLOPs: 4109464576
|
|
Parameters: 25557032
|
|
Training Data: ImageNet-1k
|
|
In Collection: SimCLR
|
|
Results:
|
|
- Task: Image Classification
|
|
Dataset: ImageNet-1k
|
|
Metrics:
|
|
Top 1 Accuracy: 69.2
|
|
Weights: https://download.openmmlab.com/mmselfsup/1.x/simclr/simclr_resnet50_16xb256-coslr-800e_in1k/resnet50_linear-8xb512-coslr-90e_in1k/resnet50_linear-8xb512-coslr-90e_in1k_20220825-b80ae1e5.pth
|
|
Config: configs/simclr/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py
|