ImpactSeg / body /Prediction.yml
VBoussot's picture
Fix model-intrinsic params in the model (input channels / class count / network architecture) instead of exposing them in the config
04f9b85 verified
Raw
History Blame Contribute Delete
1.75 kB
Predictor:
Model:
classpath: Model:ResEnc
ResEnc:
outputs_criterions: None
Dataset:
groups_src:
Volume_0:
groups_dest:
Volume:
transforms:
TensorCast:
dtype: float32
inverse: false
ResampleToResolution:
spacing:
- 3
- 3
- 3
inverse: true
Standardize:
lazy: false
mean: None
std: None
mask: None
inverse: false
patch_transforms: None
is_input: true
augmentations: None
Patch:
patch_size:
- 1
- 192
- 192
overlap: None
mask: None
pad_value: -2
extend_slice: 4
subset: None
filter: None
dataset_filenames:
- ./Dataset/:mha
use_cache: false
batch_size: 32
num_workers: None
pin_memory: false
prefetch_factor: None
persistent_workers: None
outputs_dataset:
DecoderOutputs:
OutputDataset:
name_class: OutSameAsGroupDataset
before_reduction_transforms: None
after_reduction_transforms: None
final_transforms:
Softmax:
dim: 0
Argmax:
dim: 0
TensorCast:
dtype: uint8
inverse: true
dataset_filename: Output:mha
group: Output
same_as_group: Volume_0:Volume
patch_combine: None
inverse_transform: true
reduction: Mean
Mean: {}
train_name: ImpactSeg-Body
manual_seed: 42
gpu_checkpoints: None
images_log: None
combine: Mean
autocast: false
data_log: None