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()

模型.summary()

Answer 1

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

model.summary(line_length = 100)