I’m not aware of training imitations of mobile (and thought training is disabled by default there), but your loss curve looks as if the learning rate might be too high and a “bad” update is throwing the parameters off.