/torchnnprofiler

Context Manager to profile the forward and backward times of PyTorch's nn.Module

Primary LanguagePythonMIT LicenseMIT

No issues in this repository yet.