Trying some callbacks to try to make model thinner.