VBoussot commited on
Commit
eab2ec2
·
verified ·
1 Parent(s): 111de7a

Update total/Prediction.yml

Browse files
Files changed (1) hide show
  1. total/Prediction.yml +2 -2
total/Prediction.yml CHANGED
@@ -73,7 +73,8 @@ Predictor:
73
  Head:Softmax:
74
  OutputDataset:
75
  name_class: OutSameAsGroupDataset
76
- before_reduction_transforms:
 
77
  Softmax:
78
  dim: 0
79
  Argmax:
@@ -81,7 +82,6 @@ Predictor:
81
  TensorCast:
82
  dtype: uint8
83
  inverse: true
84
- after_reduction_transforms: None
85
  final_transforms: None
86
  dataset_filename: Dataset:mha
87
  group: Output
 
73
  Head:Softmax:
74
  OutputDataset:
75
  name_class: OutSameAsGroupDataset
76
+ before_reduction_transforms: None
77
+ after_reduction_transforms:
78
  Softmax:
79
  dim: 0
80
  Argmax:
 
82
  TensorCast:
83
  dtype: uint8
84
  inverse: true
 
85
  final_transforms: None
86
  dataset_filename: Dataset:mha
87
  group: Output