GitHub repository: kshitij12345/torchnnprofiler
Primary language: Python
Created: 2022-10-22T10:35:00Z
Description: Context Manager to profile the forward and backward times of PyTorch's nn.Module