change hyperparmeter search to use MSE instead of MAE

configs/hparams_search/{focus150_optuna.yaml → focusMSE_150_optuna.yaml}

@@ -12,7 +12,7 @@ defaults:
 # make sure this is the correct name of some metric logged in lightning module!
 optimized_metric: "val/mae_best"
 
-name: "focus150_optuna"
+name: "focusMSE_150_hyperparameter_search"
 
 # here we define Optuna hyperparameter search
 # it optimizes for value returned from function with @hydra.main decorator
@@ -26,7 +26,7 @@ hydra:
     storage: null
 
     # name of the study to persist optimization results
-    study_name: focus150_hyperparameter_search
+    study_name: focusMSE_150_hyperparameter_search
 
     # number of parallel workers
     n_jobs: 1
@@ -49,6 +49,8 @@ hydra:
       datamodule.batch_size:
         type: categorical
         choices: [64, 128]
+      model:
+        _target_: src.models.focus_module.FocusMSELitModule
       model.lr:
         type: float
         low: 0.0001