Add `Model.save_spec` property to get the model's call argument TensorSpecs.
This change also enables models with multiple input arguments to be saved, for example: ``` class Subclass(keras.Model): def call(self, a, b): ... ``` PiperOrigin-RevId: 381148805
Showing
想要评论请 注册 或 登录