Unable to Execute Code

[longzilicart]

Hi, thank you for your nice work. I've been trying to replicate the experiments, but I'm encountering several issues when attempting to run the code specifically in vihg.py, which contrasts with my successful execution of the original VIG.

When I attempt to run a simple test in vihg.py, I encounter multiple errors:

if __name__ == "__main__":
    net = vihg_b_224_gelu()
    x = torch.randn((2, 3, 224, 224))
    y = net(x)
    print(y.shape)

Could you please help me understand if there's anything specific I am missing, or if there might be an issue with the code.

Thank you!