changed method to 2 phase pretrain/finetune