Include Fine Tuning in the training