Keras: отображать только NN-архитектуру без выделения памяти модели ⇐ Python

[Anonymous]

Моя цель — использовать Keras для визуализации архитектуры модели. Никакого обучения, никаких выводов.
Например, если я просто хочу визуализировать граф сети классической модели VGG16 с помощью

Код: Выделить всё

model = create_vgg_like_model()

, тогда Keras начинает предварительно выделять необходимую память и происходит сбой программы (см. изображение).


Я пытался поиграть с

Код: Выделить всё

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

но безуспешно.
Есть ли способ отложить выделение памяти или получить объект типа Model, содержащий только информацию о сети (имя слоя, ввод, вывод,...)?

Подробнее здесь: https://stackoverflow.com/questions/798 ... allocation