Efficient-AI-Backbones icon indicating copy to clipboard operation
Efficient-AI-Backbones copied to clipboard
verified publisher icon huawei-noah

Where is Multi-head implementation Code in ViG

Open xuyankun opened this issue 2 years ago • 2 comments

Dear authors, Did you show the Multi-head code in ViG? I did not find it. Could you help me point out where it is?

xuyankun avatar Jul 13 '23 07:07 xuyankun

It can easily implemented by setting groups in https://github.com/huawei-noah/Efficient-AI-Backbones/blob/524c8d7eb610b06de179cb6235d44c99581292fe/vig_pytorch/gcn_lib/torch_nn.py#L58

iamhankai avatar Jul 16 '23 14:07 iamhankai

It can easily implemented by setting groups in

https://github.com/huawei-noah/Efficient-AI-Backbones/blob/524c8d7eb610b06de179cb6235d44c99581292fe/vig_pytorch/gcn_lib/torch_nn.py#L58

Sincerely thanks for such quick reply!

xuyankun avatar Jul 19 '23 14:07 xuyankun