Count the FLOPs & Params of your OneFlow model. - View it on GitHub