train full model in the beginning