Zählen Sie die Anzahl der trainierbaren Parameter vom Modellstart bis zur angegebenen Ebene ⇐ Python

[Anonymous]

Ich habe eine Reihe von Pytorch-Modellen und möchte die Anzahl der trainierbaren Parameter ausdrucken, die erforderlich waren, um zu bestimmten Punkten in der Feed-Forward-Funktion zu gelangen.
Ich weiß, wie ich die Zahl bekomme aller trainierbaren Parameter, aber gibt es eine Möglichkeit, nur die Parameter eines Teils des Modells abzurufen? Ich könnte neue Modelle definieren, die nur bis zu einem bestimmten Punkt Ebenen haben, aber dies manuell zu tun wäre für viele Modelle zeitaufwändig.