Understanding Training Dynamics in Deep Learning using Simplified Models