SFDA: Fixing broken reliance on train.initialize_model to load and restore the model's weights.
SFDA: Fixing broken reliance on train.initialize_model to load and restore the model's weights.

PiperOrigin-RevId: 484531316

google-research/chirp