我有一个模型,它由几个继承自tf.keras.Model的子模型组成。这些子模型或多或少都是组成keras.layers的keras.layers模型(如keras.layers.Conv2D、keras.layers.BatchNormalization等)的简单集合。调用函数通过不同的顺序模型传递数据(有时向顺序模型的输出添加额外的内容,即la a ResidualBlock子模型)。
我的主要模型由子模型组成的原因是因为主模型很复杂,这样我就可以很容易地改变模型体系结构(例如子模型A的层数)。此外,子模型的某些部分实例化某些层(如keras.layers.Reshape) in the