Where is Multi-head implementation Code in ViG

[xuyankun]

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?

[iamhankai]

It can easily implemented by setting groups in

Efficient-AI-Backbones/vig_pytorch/gcn_lib/torch_nn.py

Line 58 in 524c8d7

m.append(Conv2d(channels[i - 1], channels[i], 1, bias=bias, groups=4))

[xuyankun]

It can easily implemented by setting groups in

Efficient-AI-Backbones/vig_pytorch/gcn_lib/torch_nn.py

Line 58 in 524c8d7

m.append(Conv2d(channels[i - 1], channels[i], 1, bias=bias, groups=4))

Sincerely thanks for such quick reply!