LiyuanLucasLiu / torchprof

PyTorch layer-by-layer model profiler

https://pypi.org/project/torchprof/

Geek Repo

Github PK Tool