diff --git a/pretrained_models/config.yaml b/pretrained_models/config.yaml
index 341196ea..706451f3 100644
--- a/pretrained_models/config.yaml
+++ b/pretrained_models/config.yaml
@@ -1,24 +1,24 @@
 model:
-    type: DeepSet
-    model_collection: agg_models
-    checkpoint: combined_agg.pt
-    config:
-        phi_layers: 2
-        phi_hidden_dim: 20
-        rho_layers: 3
-        rho_hidden_dim: 10
-        activation: LeakyReLU
-        pool: max
-        use_sigmoid: True
-        metrics:
-            objective: MSE
-            objective_mode: min
-            loss: MSE
-            all:
-                MSE: {}
-                PearsonCorrTorch: {}
-                MAE: {}
-                RSquared: {}
-        optimizer:
-            type: AdamW
-            config: {}
+  checkpoint: combined_agg.pt
+  config:
+    activation: LeakyReLU
+    metrics:
+      all:
+        MAE: {}
+        MSE: {}
+        PearsonCorrTorch: {}
+        RSquared: {}
+      loss: MSE
+      objective: MSE
+      objective_mode: min
+    optimizer:
+      config: {}
+      type: AdamW
+    phi_hidden_dim: 20
+    phi_layers: 2
+    pool: max
+    rho_hidden_dim: 10
+    rho_layers: 3
+    use_sigmoid: true
+  model_collection: agg_models
+  type: DeepSet
diff --git a/pretrained_models/models/repeat_0/best/bag_0.ckpt b/pretrained_models/models/repeat_0/best/bag_0.ckpt
new file mode 100644
index 00000000..942fea37
Binary files /dev/null and b/pretrained_models/models/repeat_0/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_1/best/bag_0.ckpt b/pretrained_models/models/repeat_1/best/bag_0.ckpt
new file mode 100644
index 00000000..de2a068d
Binary files /dev/null and b/pretrained_models/models/repeat_1/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_1/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_1/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_10/best/bag_0.ckpt b/pretrained_models/models/repeat_10/best/bag_0.ckpt
new file mode 100644
index 00000000..e45588e7
Binary files /dev/null and b/pretrained_models/models/repeat_10/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_11/best/bag_0.ckpt b/pretrained_models/models/repeat_11/best/bag_0.ckpt
new file mode 100644
index 00000000..79b79729
Binary files /dev/null and b/pretrained_models/models/repeat_11/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_12/best/bag_0.ckpt b/pretrained_models/models/repeat_12/best/bag_0.ckpt
new file mode 100644
index 00000000..a5a3276c
Binary files /dev/null and b/pretrained_models/models/repeat_12/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_12/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_12/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_13/best/bag_0.ckpt b/pretrained_models/models/repeat_13/best/bag_0.ckpt
new file mode 100644
index 00000000..ff5c6622
Binary files /dev/null and b/pretrained_models/models/repeat_13/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_14/best/bag_0.ckpt b/pretrained_models/models/repeat_14/best/bag_0.ckpt
new file mode 100644
index 00000000..382ca448
Binary files /dev/null and b/pretrained_models/models/repeat_14/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_15/best/bag_0.ckpt b/pretrained_models/models/repeat_15/best/bag_0.ckpt
new file mode 100644
index 00000000..f0f7c5d9
Binary files /dev/null and b/pretrained_models/models/repeat_15/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_16/best/bag_0.ckpt b/pretrained_models/models/repeat_16/best/bag_0.ckpt
new file mode 100644
index 00000000..d203c7d9
Binary files /dev/null and b/pretrained_models/models/repeat_16/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_16/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_16/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_17/best/bag_0.ckpt b/pretrained_models/models/repeat_17/best/bag_0.ckpt
new file mode 100644
index 00000000..48af4b21
Binary files /dev/null and b/pretrained_models/models/repeat_17/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_18/best/bag_0.ckpt b/pretrained_models/models/repeat_18/best/bag_0.ckpt
new file mode 100644
index 00000000..2116fa04
Binary files /dev/null and b/pretrained_models/models/repeat_18/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_19/best/bag_0.ckpt b/pretrained_models/models/repeat_19/best/bag_0.ckpt
new file mode 100644
index 00000000..42c5e8e0
Binary files /dev/null and b/pretrained_models/models/repeat_19/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_2/best/bag_0.ckpt b/pretrained_models/models/repeat_2/best/bag_0.ckpt
new file mode 100644
index 00000000..c49384a1
Binary files /dev/null and b/pretrained_models/models/repeat_2/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_20/best/bag_0.ckpt b/pretrained_models/models/repeat_20/best/bag_0.ckpt
new file mode 100644
index 00000000..2241378a
Binary files /dev/null and b/pretrained_models/models/repeat_20/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_20/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_20/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_21/best/bag_0.ckpt b/pretrained_models/models/repeat_21/best/bag_0.ckpt
new file mode 100644
index 00000000..99c24f2b
Binary files /dev/null and b/pretrained_models/models/repeat_21/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_22/best/bag_0.ckpt b/pretrained_models/models/repeat_22/best/bag_0.ckpt
new file mode 100644
index 00000000..edc6ff36
Binary files /dev/null and b/pretrained_models/models/repeat_22/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_23/best/bag_0.ckpt b/pretrained_models/models/repeat_23/best/bag_0.ckpt
new file mode 100644
index 00000000..d8443f41
Binary files /dev/null and b/pretrained_models/models/repeat_23/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_23/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_23/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_24/best/bag_0.ckpt b/pretrained_models/models/repeat_24/best/bag_0.ckpt
new file mode 100644
index 00000000..26210be4
Binary files /dev/null and b/pretrained_models/models/repeat_24/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_25/best/bag_0.ckpt b/pretrained_models/models/repeat_25/best/bag_0.ckpt
new file mode 100644
index 00000000..9d5be90c
Binary files /dev/null and b/pretrained_models/models/repeat_25/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_25/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_25/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_26/best/bag_0.ckpt b/pretrained_models/models/repeat_26/best/bag_0.ckpt
new file mode 100644
index 00000000..581c4818
Binary files /dev/null and b/pretrained_models/models/repeat_26/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_27/best/bag_0.ckpt b/pretrained_models/models/repeat_27/best/bag_0.ckpt
new file mode 100644
index 00000000..975b62c2
Binary files /dev/null and b/pretrained_models/models/repeat_27/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_27/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_27/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_28/best/bag_0.ckpt b/pretrained_models/models/repeat_28/best/bag_0.ckpt
new file mode 100644
index 00000000..3d7667c2
Binary files /dev/null and b/pretrained_models/models/repeat_28/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_29/best/bag_0.ckpt b/pretrained_models/models/repeat_29/best/bag_0.ckpt
new file mode 100644
index 00000000..006fdb26
Binary files /dev/null and b/pretrained_models/models/repeat_29/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_3/best/bag_0.ckpt b/pretrained_models/models/repeat_3/best/bag_0.ckpt
new file mode 100644
index 00000000..acd9b964
Binary files /dev/null and b/pretrained_models/models/repeat_3/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_4/best/bag_0.ckpt b/pretrained_models/models/repeat_4/best/bag_0.ckpt
new file mode 100644
index 00000000..ec08489d
Binary files /dev/null and b/pretrained_models/models/repeat_4/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_5/best/bag_0.ckpt b/pretrained_models/models/repeat_5/best/bag_0.ckpt
new file mode 100644
index 00000000..fd6fd3cf
Binary files /dev/null and b/pretrained_models/models/repeat_5/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_5/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_5/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_6/best/bag_0.ckpt b/pretrained_models/models/repeat_6/best/bag_0.ckpt
new file mode 100644
index 00000000..aed347a7
Binary files /dev/null and b/pretrained_models/models/repeat_6/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_6/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_6/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_7/best/bag_0.ckpt b/pretrained_models/models/repeat_7/best/bag_0.ckpt
new file mode 100644
index 00000000..09312c0e
Binary files /dev/null and b/pretrained_models/models/repeat_7/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_7/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_7/best/bag_0.ckpt.reverse
new file mode 100644
index 00000000..e69de29b
diff --git a/pretrained_models/models/repeat_8/best/bag_0.ckpt b/pretrained_models/models/repeat_8/best/bag_0.ckpt
new file mode 100644
index 00000000..a6bdbc36
Binary files /dev/null and b/pretrained_models/models/repeat_8/best/bag_0.ckpt differ
diff --git a/pretrained_models/models/repeat_9/best/bag_0.ckpt b/pretrained_models/models/repeat_9/best/bag_0.ckpt
new file mode 100644
index 00000000..f23bce56
Binary files /dev/null and b/pretrained_models/models/repeat_9/best/bag_0.ckpt differ