# Inference-only config (active: 4x).
# Other validation rates are commented under inference.dataset.transforms.masking.
# Pair with checkpoint vsharp_fixed_1d_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:
    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:
                    - 4.0327
                center_fractions:
                    - 0.04
                # --- 6x: uncomment these; comment out the active lists above ---
                # accelerations: [6]
                # center_fractions: [0.04]
                # --- 8x: uncomment these; comment out the active lists above ---
                # accelerations: [8.2]
                # center_fractions: [0.04]
            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
            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