Training the baseline model