KyanChen / MakeMultiHeadNaive

Use naive MultiheadAttention implement to replace nn.MultiheadAttention in pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

KyanChen/MakeMultiHeadNaive Stargazers