search

zugexiaodui / torch_flops

A library for calculating the FLOPs in the forward() process based on torch.fx
MIT License
87 stars 2 forks source link

add some funcs #8

Closed Liar-zzy closed 10 months ago

Liar-zzy commented 10 months ago

add some funcions.

Liar-zzy commented 10 months ago

@zugexiaodui please check this pr