Flexibly track outputs and grad-outputs of torch.nn.Module.
Home Page:https://graphcore-research.github.io/pytorch-tensor-tracker/
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool