adding getter for model 1 params and added pretraining for model 2
made script take argument for which model to use