metafile.yml 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. Collections:
  2. - Name: SimCLR
  3. Metadata:
  4. Training Data: ImageNet-1k
  5. Training Techniques:
  6. - LARS
  7. Training Resources: 8x V100 GPUs (b256), 16x A100-80G GPUs (b4096)
  8. Architecture:
  9. - ResNet
  10. - SimCLR
  11. Paper:
  12. Title: A simple framework for contrastive learning of visual representations
  13. URL: https://arxiv.org/abs/2002.05709
  14. README: configs/simclr/README.md
  15. Models:
  16. - Name: simclr_resnet50_16xb256-coslr-200e_in1k
  17. Metadata:
  18. Epochs: 200
  19. Batch Size: 4096
  20. FLOPs: 4109364224
  21. Parameters: 27968832
  22. Training Data: ImageNet-1k
  23. In Collection: SimCLR
  24. Results: null
  25. Weights: https://download.openmmlab.com/mmselfsup/1.x/simclr/simclr_resnet50_16xb256-coslr-200e_in1k/simclr_resnet50_16xb256-coslr-200e_in1k_20220825-4d9cce50.pth
  26. Config: configs/simclr/simclr_resnet50_16xb256-coslr-200e_in1k.py
  27. Downstream:
  28. - resnet50_simclr-200e-pre_8xb512-linear-coslr-90e_in1k
  29. - Name: simclr_resnet50_16xb256-coslr-800e_in1k
  30. Metadata:
  31. Epochs: 200
  32. Batch Size: 4096
  33. FLOPs: 4109364224
  34. Parameters: 27968832
  35. Training Data: ImageNet-1k
  36. In Collection: SimCLR
  37. Results: null
  38. Weights: https://download.openmmlab.com/mmselfsup/1.x/simclr/simclr_resnet50_16xb256-coslr-800e_in1k/simclr_resnet50_16xb256-coslr-800e_in1k_20220825-85fcc4de.pth
  39. Config: configs/simclr/simclr_resnet50_16xb256-coslr-800e_in1k.py
  40. Downstream:
  41. - resnet50_simclr-800e-pre_8xb512-linear-coslr-90e_in1k
  42. - Name: resnet50_simclr-200e-pre_8xb512-linear-coslr-90e_in1k
  43. Metadata:
  44. Epochs: 90
  45. Batch Size: 4096
  46. FLOPs: 4109464576
  47. Parameters: 25557032
  48. Training Data: ImageNet-1k
  49. In Collection: SimCLR
  50. Results:
  51. - Task: Image Classification
  52. Dataset: ImageNet-1k
  53. Metrics:
  54. Top 1 Accuracy: 66.9
  55. 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
  56. Config: configs/simclr/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py
  57. - Name: resnet50_simclr-800e-pre_8xb512-linear-coslr-90e_in1k
  58. Metadata:
  59. Epochs: 90
  60. Batch Size: 4096
  61. FLOPs: 4109464576
  62. Parameters: 25557032
  63. Training Data: ImageNet-1k
  64. In Collection: SimCLR
  65. Results:
  66. - Task: Image Classification
  67. Dataset: ImageNet-1k
  68. Metrics:
  69. Top 1 Accuracy: 69.2
  70. 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
  71. Config: configs/simclr/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py