pip install torch-multi-head-attentionfrom torch_multi_head_attention import MultiHeadAttention
MultiHeadAttention(in_features=768, head_num=12)
This repository was archived by the owner on Mar 3, 2024. It is now read-only.
CyberZHG/torch-multi-head-attention
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|