我有一个经过培训并保存了模型的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()
model.summary()
答案 0 :(得分:1)
您可以设置line_length
函数的tf.summary
属性。
model.summary(line_length = 100)