Cleaned up training procedure (non-curriculum learning).