zugexiaodui / torch_flops

A library for calculating the FLOPs in the forward() process based on torch.fx
99Updated 6 months ago

Alternatives and similar repositories for torch_flops:

Users that are interested in torch_flops are comparing it to the libraries listed below