metafile.yml 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. Collections:
  2. - Name: MILAN
  3. Metadata:
  4. Training Data: ImageNet-1k
  5. Training Techniques:
  6. - AdamW
  7. Training Resources: 16x A100-80G GPUs
  8. Architecture:
  9. - ViT
  10. Paper:
  11. Title: 'MILAN: Masked Image Pretraining on Language Assisted Representation'
  12. URL: https://arxiv.org/pdf/2208.06049
  13. README: configs/milan/README.md
  14. Models:
  15. - Name: milan_vit-base-p16_16xb256-amp-coslr-400e_in1k
  16. Metadata:
  17. Epochs: 400
  18. Batch Size: 4096
  19. FLOPs: 17581972224
  20. Parameters: 111907584
  21. Training Data: ImageNet-1k
  22. In Collection: MILAN
  23. Results: null
  24. Weights: https://download.openmmlab.com/mmselfsup/1.x/milan/milan_vit-base-p16_16xb256-amp-coslr-400e_in1k/milan_vit-base-p16_16xb256-amp-coslr-400e_in1k_20221129-180922e8.pth
  25. Config: configs/milan/milan_vit-base-p16_16xb256-amp-coslr-400e_in1k.py
  26. Downstream:
  27. - vit-base-p16_milan-pre_8xb128-coslr-100e_in1k
  28. - vit-base-p16_milan-pre_8xb2048-linear-coslr-100e_in1k
  29. - Name: vit-base-p16_milan-pre_8xb128-coslr-100e_in1k
  30. Metadata:
  31. Epochs: 100
  32. Batch Size: 1024
  33. FLOPs: 17581215744
  34. Parameters: 86566120
  35. Training Data: ImageNet-1k
  36. In Collection: MILAN
  37. Results:
  38. - Task: Image Classification
  39. Dataset: ImageNet-1k
  40. Metrics:
  41. Top 1 Accuracy: 85.3
  42. Weights: https://download.openmmlab.com/mmselfsup/1.x/milan/milan_vit-base-p16_16xb256-amp-coslr-400e_in1k/vit-base-p16_ft-8xb128-coslr-100e_in1k/vit-base-p16_ft-8xb128-coslr-100e_in1k-milan_20221129-74ac94fa.pth
  43. Config: configs/milan/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py
  44. - Name: vit-base-p16_milan-pre_8xb2048-linear-coslr-100e_in1k
  45. Metadata:
  46. Epochs: 100
  47. Batch Size: 16384
  48. FLOPs: 17581972992
  49. Parameters: 86567656
  50. Training Data: ImageNet-1k
  51. In Collection: MILAN
  52. Results:
  53. - Task: Image Classification
  54. Dataset: ImageNet-1k
  55. Metrics:
  56. Top 1 Accuracy: 78.9
  57. Weights: https://download.openmmlab.com/mmselfsup/1.x/milan/milan_vit-base-p16_16xb256-amp-coslr-400e_in1k/vit-base-p16_linear-8xb2048-coslr-100e_in1k/vit-base-p16_linear-8xb2048-coslr-100e_in1k_20221129-03f26f85.pth
  58. Config: configs/milan/benchmarks/vit-base-p16_8xb2048-linear-coslr-100e_in1k.py