diff --git a/didactic/config/experiment/cardinal/cardiac-multimodal-representation.yaml b/didactic/config/experiment/cardinal/cardiac-multimodal-representation.yaml index 9d07c418..fc485899 100644 --- a/didactic/config/experiment/cardinal/cardiac-multimodal-representation.yaml +++ b/didactic/config/experiment/cardinal/cardiac-multimodal-representation.yaml @@ -75,9 +75,13 @@ data: - ht_cm image_attributes: - gls + - gls_left + - gls_right - lv_area - lv_length - myo_area + - myo_area_left + - myo_area_right predict: _target_: didactic.data.cardinal.predict.CardiacRepresentationPredictionWriter @@ -85,6 +89,9 @@ data: hue_attrs: ${data.process_patient_kwargs.clinical_attributes} excluded_clinical_attrs: [] +# Determine the number of clinical + image attributes used by the model based on their respective configs +n_clinical_attrs: ${builtin.len:${task.clinical_attrs}} +n_img_attrs: ${op.mul:${builtin.len:${task.views}},${builtin.len:${task.img_attrs}}} task: _target_: didactic.tasks.cardiac_multimodal_representation.CardiacMultimodalRepresentationTask @@ -172,6 +179,7 @@ hydra: - task.attrs_dropout - task.embed_dim + - task/model - task.model.encoder.num_layers - task.model.encoder.encoder_layer.nhead - task.model.encoder.encoder_layer.dim_feedforward