VBoussot
/

ImpactSynth

medical-imaging

image-synthesis

Model card Files Files and versions

VBoussot commited on May 22

Commit

a5751b7

·

verified ·

1 Parent(s): 33b2db9

Update MR_CBCT/Prediction.yml

Files changed (1) hide show

MR_CBCT/Prediction.yml +4 -1

MR_CBCT/Prediction.yml CHANGED Viewed

@@ -10,13 +10,16 @@ Predictor:
         groups_dest:
           Volume:
             transforms:
               Clip/0:
                 min_value: -1024
                 max_value: 5000
                 save_clip_min: false
                 save_clip_max: false
                 mask: None
-              TensorCast:
                 dtype: int16
                 inverse: false
               ResampleToResolution:

         groups_dest:
           Volume:
             transforms:
+              TensorCast/0:
+                dtype: float16
+                inverse: false
               Clip/0:
                 min_value: -1024
                 max_value: 5000
                 save_clip_min: false
                 save_clip_max: false
                 mask: None
+              TensorCast/1:
                 dtype: int16
                 inverse: false
               ResampleToResolution: