python - Tensorflow 展示模型总结

Question

我有一个 UNet，我训练并保存了模型。我想查看模型摘要，但是当我使用 model.summary() 时，Output Shape 列被截断并且无法看到完整的形状。该模型是在 3D 图像上训练的，因此输出应显示 (None, shapeX, shapeY, shapeZ, num_features)。如何显示完整的输出形状？

from tensorflow.keras.models import load_model
        
model = load_model('model.h5',compile = False)
model.summary()

模型.总结()

Answer 1

您可以设置tf.summary 函数的line_length 属性。

model.summary(line_length = 100)