Add module information to the run time and memory reports

skylineprof / skyline

🏙 Interactive in-editor performance profiling, visualization, and debugging for PyTorch neural networks.

Apache License 2.0

30 stars 3 forks source link

Add module information to the run time and memory reports #43

Open geoffxy opened 4 years ago

geoffxy commented 4 years ago

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.