moved and revamped finetuning script for 3,4 layer models