Update total/Prediction.yml

total/Prediction.yml

@@ -73,8 +73,7 @@ Predictor:
     Head:Softmax:
       OutputDataset:
         name_class: OutSameAsGroupDataset
-        before_reduction_transforms: None
-        after_reduction_transforms: 
+        before_reduction_transforms: 
           Softmax:
             dim: 0
           Argmax:
@@ -82,6 +81,7 @@ Predictor:
           TensorCast:
             dtype: uint8
             inverse: true
+        after_reduction_transforms: None
         final_transforms: None
         dataset_filename: Dataset:mha
         group: Output