kshitij12345/torchnnprofiler: Context Manager to profile the forward and backward times of PyTorch's nn.Module
Category: profiling
Language: Python