Я хочу знать время выполнения каждого слоя DNN. Есть ли какой-нибудь инструмент для профилирования каждого слоя? Я знаю, что могу вставить какой-нибудь отпечаток в модель, но не уверен, что он будет точным.
Для проверки размера и параметров каждого слоя вы можете использовать сводка факела .
Чтобы вычислить время выполнения каждого слоя, вы можете использовать torchprof .
Я не знаю ни одного проекта, объединяющего обе библиотеки. Может быть, это ваша возможность (смеется)