Added a new script to do pretraining model search on gravity, as well as modifications to the
plotting scripts for the comparison figures.x
Including the hyperparameter grid search script.