Modell Zusammenfassung in pytorch
Gibt es eine Möglichkeit, ich kann drucken Sie die Zusammenfassung eines Modells in PyTorch wie model.summary()
Methode wird in Keras wie folgt?
Model Summary:
____________________________________________________________________________________________________
Layer (type) Output Shape Param # Connected to
====================================================================================================
input_1 (InputLayer) (None, 1, 15, 27) 0
____________________________________________________________________________________________________
convolution2d_1 (Convolution2D) (None, 8, 15, 27) 872 input_1[0][0]
____________________________________________________________________________________________________
maxpooling2d_1 (MaxPooling2D) (None, 8, 7, 27) 0 convolution2d_1[0][0]
____________________________________________________________________________________________________
flatten_1 (Flatten) (None, 1512) 0 maxpooling2d_1[0][0]
____________________________________________________________________________________________________
dense_1 (Dense) (None, 1) 1513 flatten_1[0][0]
====================================================================================================
Total params: 2,385
Trainable params: 2,385
Non-trainable params: 0
- Haben Sie gesehen, die state_dict () - Methode auf das Modul?? Es gibt Sie, die verschiedenen Parameter des Modells. Es gibt keine direkte Zusammenfassung Methode, aber man könnte, wie man mit dem state_dict () - Methode
Du musst angemeldet sein, um einen Kommentar abzugeben.
Während Sie nicht erhalten Sie detaillierte Informationen über das Modell, wie in Keras' - Modell.Zusammenfassung, einfach drucken Sie das Modell wird Ihnen eine Vorstellung über die verschiedenen Ebenen beteiligt und Ihre Spezifikationen.
Zum Beispiel:
Die Ausgabe in diesem Fall wäre etwas wie folgt:
Nun kann man, wie erwähnt, durch Kashyap, verwenden Sie die
state_dict
Methode, um die GEWICHTE der verschiedenen Schichten. Aber mit dieser Auflistung der Schichten würden eventuell mehr Richtung ist die Schaffung einer helper-Funktion zu bekommen, dass Keras wie Modell Zusammenfassung! Hoffe, das hilft!Ja, Sie können, get genaue Keras Darstellung, mit pytorch-Zusammenfassung Paket.
Beispiel für VGG16
Expected object of backend CPU but got backend CUDA for argument #4 'mat1'
und es scheint, dass es keine Dokumentation gibt.summary(model.cuda(), (INPUT_SHAPE))
funktioniert.Dies zeigt ein Modell die GEWICHTE und Parameter (aber nicht output-Form).
Edit: isaykatsman hat eine pytorch PR hinzufügen
model.summary()
ist das genau wie keras https://github.com/pytorch/pytorch/pull/3043/filesEinfachsten zu merken (nicht so schön wie Keras):
Diese arbeiten auch:
Wenn Sie wollen einfach nur die Anzahl der Parameter:
Aus: Gibt es ähnliche pytorch Funktion als Modell.Zusammenfassung() als keras? (forum.PyTorch.org)
AFAK es ist kein Modell.Zusammenfassung() wie äquivalent in pytorch
Mittlerweile finden Sie Skript von szagoruyko, das gibt einen schönen visualizaton wie in resnet18-Beispiel
Cheers
Um torchsummary wie diese:
Aber müssen Sie es zuerst installieren, wenn es nicht bereits vorhanden ist.
Und dann kann man es versuchen, aber nicht aus irgendeinem Grund es nicht funktioniert, es sei denn, ich Modell cuda
alexnet.cuda
:Den
summary
müssen die input-Größe und batch-Größe wird auf -1 gesetzt, D. H. alle batch-Größe, die wir bieten.Wenn wir
summary(alexnet, (3, 224, 224), 32)
dies bedeutet, dass die Verwendung derbs=32
.Sich
Drucken Sie einfach das Modell nach der Definition ein Objekt der model-Klasse