ma-xu / pointMLP-pytorch

[ICLR 2022 poster] Official PyTorch implementation of "Rethinking Network Design and Local Geometry in Point Cloud: A Simple Residual MLP Framework"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Cite your work

xiaobaitu123344 opened this issue · comments

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.

commented

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

Thank you very much for your answer