GitHub repository: zugexiaodui/torch_flops
Primary language: Python
Created: 2023-10-26T11:56:23Z
Description: A library for calculating the FLOPs in the forward() process based on torch.fx