/torchrecorder

A small package to record execution graphs of PyTorch neural networks

Primary LanguagePythonMIT LicenseMIT

Issues