corrected the output directory for 3,5 layer pretraining scripts