"tests/kernels/conftest.py" did not exist on "22379d55130cb3c8d8c2130eedc0eaedf5e209e1"
-
Zhichao Lu authored
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
93b8168a