mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1406 from cuicheng01/release/2.3

Browse Source 
【cherry-pick】update vision_transformer.py
This commit is contained in:
cuicheng01 2021-11-05 11:09:33 +08:00 committed by GitHub
commit 4428868ea7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/arch/backbone/model_zoo/vision_transformer.py
View File

@ -12,6 +12,8 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # 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 from collections.abc import Callable
import numpy as np import numpy as np