Add warmstart and only doing fine-tuning parameters for testing

Adding warm-start which is either always start with pretraining or start from the last budget level. Also adding a parameter to switch MME to only fine-tuning.