diff --git a/trained_model_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml b/trained_model_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml
index 68bfb41e..cae1ac81 100644
--- a/trained_model_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml
+++ b/trained_model_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml
@@ -243,6 +243,6 @@ predict:
     _target_: myria3d.models.interpolation.Interpolator
     interpolation_k: 10
     classification_dict: ${dataset_description.classification_dict}
-    probas_to_save: all
-    predicted_classification_channel: PredictedClassification
+    probas_to_save: [building,ground]
+    predicted_classification_channel: reliability
     entropy_channel: entropy