From a1cac15258c582bac126ef8b54d39c4f984d5a33 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Fri, 5 Nov 2021 02:53:10 +0000 Subject: [PATCH] update vision_transformer.py --- ppcls/arch/backbone/model_zoo/vision_transformer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ppcls/arch/backbone/model_zoo/vision_transformer.py b/ppcls/arch/backbone/model_zoo/vision_transformer.py index 8bcfda02e..5aaf8cc35 100644 --- 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