Finetune non final ckpts

[ElanaPearl]

• add ability to specify if you want to use final checkpoint or not
• if not using final checkpoint, assumes the pretrain_dir is the checkpoint dir