Switch line orders in trainer so that restore_map is called after moving average variables are created.
Switch line orders in trainer so that restore_map is called after moving average variables are created.

Moving averages are now properly loaded during fine-tuning, instead of being recreated.

PiperOrigin-RevId: 190496046

tombstone/models,alexgorban/models,cshallue/models,cshallue/models,derekjchow/models,derekjchow/models,derekjchow/models,cshallue/models,alexgorban/models,derekjchow/models,cshallue/models,tombstone/models,tombstone/models,derekjchow/models,alexgorban/models,alexgorban/models,alexgorban/models,tombstone/models,tombstone/models,tombstone/models,cshallue/models