Update total-3mm/Prediction.yml

total-3mm/Prediction.yml

@@ -72,8 +72,7 @@ Predictor:
     Head:Softmax:
       OutputDataset:
         name_class: OutSameAsGroupDataset
-        before_reduction_transforms: None
-        after_reduction_transforms: 
+        before_reduction_transforms: 
           Softmax:
             dim: 0
           Argmax:
@@ -81,9 +80,10 @@ Predictor:
           TensorCast:
             dtype: uint8
             inverse: true
+        after_reduction_transforms: None
         final_transforms: None
         dataset_filename: Dataset:mha
-        group: Seg
+        group: Output
         same_as_group: Volume:Volume
         patch_combine: Cosinus
         inverse_transform: true