torchscope

torchscope copied to clipboard

As a lazy coder ,your code help me to save time and i also can learn many novel ways。

zyfone

Use jit tracing functionality for more accurate capture of all ops

1

Current impl based on forward hook only captures operations in the model that have been placed in Modules. If functional ops are used, they are missed. I believe a solution...

rwightman

I think FLOPs ≈ 2 * MAdds as general each MAdd contains one multiplication and one addition, but the result of your code is not like this.

earthkerwin