你的 mid_layer的实现对我挺有帮助的
summary 那个问题,看不到的原因是「输入」对于初始化后的模型仍然是不可知的shape,可以在你的SubclassModel里重载sumary方法如下:
def summary(self):
# 也可以用summary的参数传进来这个init_shape
x = tf.keras.Input(shape=(self.init_shape))
tf.keras.Model(inputs=[x], outputs=self.call(x)).summary()
tensorflow2.0(3)-Resnet模型tensorflow2不再需要静态建图启动session(),抛弃很多繁杂的功能设计,代码上更加简洁清晰,而在工程上也更加灵活。但是一些基础的用法,单靠api接口去训练模...