[x] Your code builds clean without any errors or warnings
[x] You are using approved terminology
[ ] You have added unit tests
Motivation
Previous ViT mainly aimed at image classification, which is not convenient to extend for unsupervised learning or detection. This PR makes ViT more extendable.
Thank you for your contribution to the MindCV repo. Before submitting this PR, please make sure:
Motivation
Previous ViT mainly aimed at image classification, which is not convenient to extend for unsupervised learning or detection. This PR makes ViT more extendable.
Test Plan
Related Issues and PRs
(Is this PR part of a group of changes? Link the other relevant PRs and Issues here. Use https://help.github.com/en/articles/closing-issues-using-keywords for help on GitHub syntax)