python - 为 Dataloader 分析 PyTorch 自定义数据集

Question

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

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

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

谢谢！