metafile.yml 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. Collections:
  2. - Name: BEiT
  3. Metadata:
  4. Architecture:
  5. - Attention Dropout
  6. - Convolution
  7. - Dense Connections
  8. - Dropout
  9. - GELU
  10. - Layer Normalization
  11. - Multi-Head Attention
  12. - Scaled Dot-Product Attention
  13. - Tanh Activation
  14. Paper:
  15. Title: 'BEiT: BERT Pre-Training of Image Transformers'
  16. URL: https://arxiv.org/abs/2106.08254
  17. README: configs/beit/README.md
  18. Code:
  19. URL: https://github.com/open-mmlab/mmpretrain/blob/main/mmpretrain/models/backbones/beit.py
  20. Version: v1.0.0rc4
  21. Models:
  22. - Name: beit_beit-base-p16_8xb256-amp-coslr-300e_in1k
  23. Metadata:
  24. Epochs: 300
  25. Batch Size: 2048
  26. FLOPs: 17581219584
  27. Parameters: 86530984
  28. Training Data: ImageNet-1k
  29. In Collection: BEiT
  30. Results: null
  31. Weights: https://download.openmmlab.com/mmselfsup/1.x/beit/beit_vit-base-p16_8xb256-amp-coslr-300e_in1k/beit_vit-base-p16_8xb256-amp-coslr-300e_in1k_20221128-ab79e626.pth
  32. Config: configs/beit/beit_beit-base-p16_8xb256-amp-coslr-300e_in1k.py
  33. Downstream:
  34. - beit-base-p16_beit-pre_8xb128-coslr-100e_in1k
  35. - Name: beit-base-p16_beit-pre_8xb128-coslr-100e_in1k
  36. Metadata:
  37. Epochs: 100
  38. Batch Size: 1024
  39. FLOPs: 17581219584
  40. Parameters: 86530984
  41. Training Data: ImageNet-1k
  42. In Collection: BEiT
  43. Results:
  44. - Task: Image Classification
  45. Dataset: ImageNet-1k
  46. Metrics:
  47. Top 1 Accuracy: 83.1
  48. Weights: https://download.openmmlab.com/mmselfsup/1.x/beit/beit_vit-base-p16_8xb256-amp-coslr-300e_in1k/vit-base-p16_ft-8xb128-coslr-100e_in1k/vit-base-p16_ft-8xb128-coslr-100e_in1k_20221128-0ca393e9.pth
  49. Config: configs/beit/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py
  50. - Name: beit-base-p16_beit-in21k-pre_3rdparty_in1k
  51. Metadata:
  52. FLOPs: 17581219584
  53. Parameters: 86530984
  54. Training Data:
  55. - ImageNet-21k
  56. - ImageNet-1k
  57. In Collection: BEiT
  58. Results:
  59. - Dataset: ImageNet-1k
  60. Task: Image Classification
  61. Metrics:
  62. Top 1 Accuracy: 85.28
  63. Top 5 Accuracy: 97.59
  64. Weights: https://download.openmmlab.com/mmclassification/v0/beit/beit-base_3rdparty_in1k_20221114-c0a4df23.pth
  65. Config: configs/beit/benchmarks/beit-base-p16_8xb64_in1k.py
  66. Converted From:
  67. Weights: https://conversationhub.blob.core.windows.net/beit-share-public/beit/beit_base_patch16_224_pt22k_ft22kto1k.pth
  68. Code: https://github.com/microsoft/unilm/tree/master/beit