direct-modulated-convolution / prostate /vsharp_modconv_features_triang.yaml
georgeyiasemis's picture
Fix inference YAMLs: pin single R+ACS; document training schedule + how to override
ac88ae7 verified
Raw
History Blame Contribute Delete
1.65 kB
model:
model_name: vsharp.vsharp.VSharpNet
num_steps: 12
num_steps_dc_gd: 10
image_init: SENSE
no_parameter_sharing: true
auxiliary_steps: -1
image_model_architecture: UNET
initializer_channels:
- 32
- 32
- 64
- 64
initializer_dilations:
- 1
- 1
- 2
- 4
initializer_multiscale: 1
initializer_activation: PRELU
conv_modulation: FEATURES
aux_in_features: 2
log_aux: true
fc_hidden_features:
- 32
- 32
fc_groups: 1
fc_activation: SOFTPLUS
modulation_at_input: false
image_unet_num_filters: 32
image_unet_num_pool_layers: 4
image_unet_dropout: 0.0
image_unet_norm_type: INSTANCE
additional_models:
sensitivity_model:
model_name: unet.unet_2d.UnetModel2d
in_channels: 2
out_channels: 2
num_filters: 16
num_pool_layers: 4
dropout_probability: 0.0
conv_out_bias: false
physics:
forward_operator: fft2
backward_operator: ifft2
inference:
batch_size: 1
crop: header
dataset:
name: FastMRI
transforms:
use_seed: true
delete_kspace: false
masking:
name: FastMRIEquispaced
accelerations:
- 4
center_fractions:
- 0.08
cropping:
image_center_crop: false
sensitivity_map_estimation:
estimate_sensitivity_maps: true
normalization:
scaling_key: masked_kspace
scale_percentile: 0.995