Last released Sep 5, 2024
A library for calculating the FLOPs in the forward() process based on torch.fx
Supported by