Cite your work

[xiaobaitu123344]

Thank you very much for your creative work, I would like to cite your paper, but have encountered a small problem about how to use from torchstat import stat to calculate FLOPs.

[ma-xu]

We didn't report the flops because some operations are implemented by cuda. You can try https://github.com/facebookresearch/fvcore

[xiaobaitu123344]

Thank you very much for your answer