update vision_transformer.py

2021-11-05 02:53:10 +00:00 · 2021-11-05 02:53:10 +00:00 · 7875e83d9e
parent 08952deb88
commit 7875e83d9e
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