skyline
skyline copied to clipboard
Add module information to the run time and memory reports
We now have module information about each operation/memory entry (i.e. the torch.nn.Module
related to each entry). This information is currently not included in the report files saved by skyline time
and skyline memory
because we do not need it. However, it would be nice to include it.