Added capabilty for different pretraining scripts for specifying final layer size.