Dataloader 的 PyTorch 自定义数据集分析

Question

我实现了一个自定义 PyTorch 数据集，供项目的数据加载器使用。然而，它的运行速度比预期慢，因此选择分析来解决瓶颈问题。

研究了 vprof，但如果我只对分析数据集实现感兴趣，我不确定要分析的函数。在 Ubuntu 18.04 上使用 PyTorch 1.5 和 Python 3.7。

我们如何仅对自定义数据集实现执行性能分析（CPU 和内存）？

谢谢！