change folder structure

Author: achiefa

n3fit/src/n3fit/model_trainer.py

@@ -12,6 +12,7 @@
 from collections import namedtuple
 from itertools import zip_longest
 import logging
+import pickle
 
 import numpy as np
 
@@ -742,7 +743,10 @@ def _train_and_fit(self, training_model, stopping_object, epochs=100) -> bool:
 
         if self.save_checkpoints:
             pdf_model = training_model.get_layer("PDFs")
-            replica_paths = [self.replica_path / f"replica_{r}" for r in self.replicas]
+            # Save parameters where colibri will look for checkpoints
+            replica_paths = [
+                self.replica_path.parent / f"fit_replicas/replica_{r}" for r in self.replicas
+            ]
             checpoint_callback = callbacks.StoreCallback(
                 pdf_model=pdf_model, replica_paths=replica_paths, check_freq=self.checkpoint_freq
             )
@@ -961,9 +965,8 @@ def hyperparametrizable(self, params):
             "layer_type": params["layer_type"],
         }
         state = {"_init_args": _init_args}
-        import pickle
 
-        with open(self.replica_path / "pdf_model.pkl", "wb") as file:
+        with open(self.replica_path.parent / "pdf_model.pkl", "wb") as file:
             pickle.dump(state, file)
 
         ### Training loop