The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)
Hi,
I was wondering how i could you your implementation of flops counter for GNN layers, in the description it says that it works for GCN but i was wondering what should be the input shape.
MrYxJ
commented
2 months ago
Hi, I was wondering how i could you your implementation of flops counter for GNN layers, in the description it says that it works for GCN but i was wondering what should be the input shape.
Thank you