vai_p_tensorflow2 only supports Keras models created by the Functional API or the Sequential API. Subclassed models are not supported.
vai_p_tensorflow2 only supports Keras models created by the Functional API or the Sequential API. Subclassed models are not supported.