| # Inference-only config (active: 4x). | |
| # Other validation rates are commented under inference.dataset.transforms.masking. | |
| # Pair with checkpoint vsharp_ads_1d_phase_init_reg.pt (same stem as the training YAML). | |
| model: | |
| model_name: vsharp.vsharp.VSharpNet3D | |
| num_steps: 8 | |
| num_steps_dc_gd: 6 | |
| image_init: SENSE | |
| no_parameter_sharing: true | |
| auxiliary_steps: -1 | |
| initializer_channels: | |
| - 32 | |
| - 32 | |
| - 64 | |
| - 64 | |
| initializer_dilations: | |
| - 1 | |
| - 1 | |
| - 2 | |
| - 4 | |
| initializer_multiscale: 1 | |
| initializer_activation: PRELU | |
| unet_num_filters: 16 | |
| unet_num_pool_layers: 4 | |
| unet_dropout: 0.0 | |
| unet_norm: false | |
| additional_models: | |
| sampling_model: | |
| model_name: adaptive.policy.StraightThroughPolicy | |
| sampling_dimension: ONE_D | |
| st_slope: 10.0 | |
| st_clamp: false | |
| use_softplus: true | |
| slope: 10.0 | |
| fix_sign_leakage: true | |
| kspace_shape: | |
| - 512 | |
| - 246 | |
| num_layers: 1 | |
| kspace_sampler: false | |
| sampler_detach_mask: false | |
| sampler_chans: 16 | |
| sampler_num_pool_layers: 3 | |
| sampler_fc_size: 128 | |
| sampler_drop_prob: 0.0 | |
| sampler_num_fc_layers: 3 | |
| sampler_activation: LEAKY_RELU | |
| sampling_type: DYNAMIC_2D | |
| num_time_steps: 11 | |
| 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 | |
| registration_model: | |
| model_name: registration.registration.UnetRegistration2dModel | |
| warp_num_integration_steps: 2 | |
| max_seq_len: 11 | |
| unet_num_filters: 16 | |
| unet_num_pool_layers: 4 | |
| unet_dropout_probability: 0.0 | |
| unet_normalized: true | |
| physics: | |
| forward_operator: fft2 | |
| backward_operator: ifft2 | |
| inference: | |
| batch_size: 1 | |
| dataset: | |
| name: CMRxRecon | |
| transforms: | |
| use_seed: true | |
| delete_kspace: false | |
| masking: | |
| name: FastMRIEquispaced | |
| accelerations: | |
| - 8.2 | |
| center_fractions: | |
| - 0.04 | |
| # --- 6x: uncomment these; comment out the active lists above --- | |
| # accelerations: [10.25] | |
| # center_fractions: [0.04] | |
| # and set target_acceleration: 6 | |
| # --- 8x: uncomment these; comment out the active lists above --- | |
| # accelerations: [12.3] | |
| # center_fractions: [0.04] | |
| # and set target_acceleration: 8.2 | |
| mode: DYNAMIC | |
| cropping: | |
| image_center_crop: false | |
| sensitivity_map_estimation: | |
| estimate_sensitivity_maps: true | |
| normalization: | |
| scaling_key: acs_kspace | |
| scale_percentile: 0.995 | |
| use_acs_as_mask: false | |
| delete_acs_mask: false | |
| padding_eps: 0 | |
| target_acceleration: 4.0327 | |
| registration: | |
| registration: true | |
| registration_simulate_reference: FROM_KEY | |
| registration_simulate_reference_from_key_index: 6 | |
| registration_estimate_displacement: false | |
| augmentation: | |
| pad: | |
| - 512 | |
| - 246 | |
| kspace_context: time | |