Keras: Nur NN-Architektur ohne Modellspeicherzuweisung anzeigen ⇐ Python

[Anonymous]

Mein Ziel ist es, Keras zu verwenden, um die Architektur des Modells zu visualisieren. Kein Training, keine Inferenz.
Zum Beispiel, wenn ich nur den Graphen des Netzwerks des klassischen VGG16-Modells mit
visualisieren möchte

Code: Select all

model = create_vgg_like_model()

Dann beginnt Keras, den benötigten Speicher vorab zuzuweisen, und das Programm stürzt ab (siehe Bild).


Ich habe versucht, damit herumzuspielen

Code: Select all

cpu_devices = tf.config.experimental.list_physical_devices('CPU')

if cpu_devices:
try:
tf.config.experimental.set_visible_devices(cpu_devices)
tf.config.experimental.set_memory_growth(cpu_devices[0], True)
except:
pass

aber ohne Erfolg.
Gibt es entweder eine Möglichkeit, die Speicherzuweisung zu verschieben oder ein Modell-ähnliches Objekt zu erhalten, das nur die Informationen des Netzwerks enthält (Ebenenname, Eingabe, Ausgabe, ...)?