After some investigation, I realized that since the subclassed model is just a wrapper for training two functional models, I need to:

1. Explicitly compile the functional models inside the wrapper to save their training configurations.
2. Override the call function in the wrapper model and explicitly invoke it before saving the subclassed model.

I hope this helps anyone facing the same issue in the future.