PyTorch подсчитывает количество обучаемых параметров от начала модели до заданного слоя ⇐ Python

[Anonymous]

У меня есть несколько моделей pytorch, и я хотел бы напечатать количество обучаемых параметров, необходимых для достижения определенных точек в функции прямой связи.
Я знаю, как получить число всех обучаемых параметров, но есть ли способ получить параметры только части модели. Я мог бы определить новые модели, которые имеют слои только до определенной точки, но для многих моделей это вручную заняло бы много времени.

Подробнее здесь: https://stackoverflow.com/questions/793 ... iven-layer