Я реализовал собственный набор данных PyTorch для Dataloader моего проекта. Однако он работает медленнее, чем ожидалось, поэтому для устранения узкого места было выбрано профилирование.
Посмотрел vprof, но я не уверен в профилировании функции, если меня интересует только профилирование реализации набора данных. Использование PyTorch 1.5 с Python 3.7 в Ubuntu 18.04.
Как мы можем выполнить профилирование производительности (ЦП и память) только для пользовательской реализации набора данных?
Спасибо!