Merge pull request #1407 from cuicheng01/develop

update vision_transformer.py
2021-11-05 11:09:44 +08:00 · 2021-11-05 11:09:44 +08:00 · d8fbc1b36b
parent 5aba5f898e 7875e83d9e
commit d8fbc1b36b
1 changed files with 2 additions and 0 deletions
--- a/ppcls/arch/backbone/model_zoo/vision_transformer.py
+++ b/ppcls/arch/backbone/model_zoo/vision_transformer.py
@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.

+# Code was based on https://github.com/rwightman/pytorch-image-models/blob/master/timm/models/vision_transformer.py
+
 from collections.abc import Callable

 import numpy as np