deepinsight / deepinsight/insightface

About Mixed Precision in ViT.

Open
#2,010 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
29.7k
Forks
6.1k
PR merge metrics
No merged PRs in 30d

Description

Hi, I found some engineering tricks in your ViT backbone code([ViT](https://github.com/deepinsight/insightface/blob/master/recognition/arcface_torch/backbones/vit.py)).

in Attention Module

![image](https://user-images.githubusercontent.com/94514604/169200241-ad7ca3ce-a06c-4230-816c-4d205d448df2.png)

in Block Module
![image](https://user-images.githubusercontent.com/94514604/169200310-eaceab77-3dd0-4d37-b33d-2a266916c2ea.png)

in forward_features of ViT
![image](https://user-images.githubusercontent.com/94514604/169200392-2f8aa8fd-2561-4783-9586-b648929f642e.png)

Could you explain the reason of using these codes? Thanks.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.