metafile.yml 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. Collections:
  2. - Name: MaskFeat
  3. Metadata:
  4. Training Data: ImageNet-1k
  5. Training Techniques:
  6. - AdamW
  7. Training Resources: 8x A100-80G GPUs
  8. Architecture:
  9. - ViT
  10. Paper:
  11. Title: Masked Feature Prediction for Self-Supervised Visual Pre-Training
  12. URL: https://arxiv.org/abs/2112.09133v1
  13. README: configs/maskfeat/README.md
  14. Models:
  15. - Name: maskfeat_vit-base-p16_8xb256-amp-coslr-300e_in1k
  16. Metadata:
  17. Epochs: 300
  18. Batch Size: 2048
  19. FLOPs: 17581972224
  20. Parameters: 85882692
  21. Training Data: ImageNet-1k
  22. In Collection: MaskFeat
  23. Results: null
  24. Weights: https://download.openmmlab.com/mmselfsup/1.x/maskfeat/maskfeat_vit-base-p16_8xb256-amp-coslr-300e_in1k/maskfeat_vit-base-p16_8xb256-amp-coslr-300e_in1k_20221101-6dfc8bf3.pth
  25. Config: configs/maskfeat/maskfeat_vit-base-p16_8xb256-amp-coslr-300e_in1k.py
  26. Downstream:
  27. - vit-base-p16_maskfeat-pre_8xb256-coslr-100e_in1k
  28. - Name: vit-base-p16_maskfeat-pre_8xb256-coslr-100e_in1k
  29. Metadata:
  30. Epochs: 100
  31. Batch Size: 2048
  32. FLOPs: 17581215744
  33. Parameters: 86566120
  34. Training Data: ImageNet-1k
  35. In Collection: MaskFeat
  36. Results:
  37. - Task: Image Classification
  38. Dataset: ImageNet-1k
  39. Metrics:
  40. Top 1 Accuracy: 83.4
  41. Weights: https://download.openmmlab.com/mmselfsup/1.x/maskfeat/maskfeat_vit-base-p16_8xb256-amp-coslr-300e_in1k/vit-base-p16_ft-8xb256-coslr-100e_in1k/vit-base-p16_ft-8xb256-coslr-100e_in1k_20221028-5134431c.pth
  42. Config: configs/maskfeat/benchmarks/vit-base-p16_8xb256-coslr-100e_in1k.py