Flexibly track outputs and grad-outputs of torch.nn.Module.
Primary LanguagePythonMIT LicenseMIT
No issues in this repository yet.