cleaned up pretrain/finetune flow