base_config.yaml

# pipeline project
pipeline_project: !!str "ECG_Tokenizer_Linear_Probing"
model_name: !!str "ECG_Tokenizer_Wrapper"
runner_name: !!str "ECG_Tokenizer_Linear_Probing_Runner"
run_mode: !!str "inference"
num_epochs: !!int 10
seed: !!int 42
base_checkpoint_path: !!str "/app/outputs/"
pretrained_tokenizer_path: !!str "/app/checkpoints/DeepECG-Tok_tokenizer/deepecg_tokenizer.pt"
inference_dataset_path: !!str "/app/outputs/preprocessed_bert_output.parquet"
inference_checkpoint_path: !!str "/app/checkpoints/DeepECG-Tok_EfficientNetV2_77_Classes/deepecg_tokenizer_efficientnet.pt"
# wandb parameters
wandb_project: !!str "ECG_Tokenizer_Linear_Probing"
wandb_entity: !!str "mhi_ai"
use_wandb: !!bool false
# training parameters
lr: !!float 3e-4
scheduler_type: !!str "cosine_warm_restart"
lr_step_period: !!int 1
factor: !!float 0.3
optimizer: !!str "AdamW"
weight_decay: !!float 1e-4
step_size: !!int 1
gamma: !!float 0.1
num_warmup_percent: !!float 0.1
num_hard_restarts_cycles: !!float 0.5
warm_restart_tmult: !!int 2
# VQVAE parameters
decoder_mode: !!str "classification"
decoder_name: !!str "EfficientNetV2_Classifier_Decoder"
# Classification parameters
num_classes: !!int 77
criterion: !!str "bce_loss"
# Dataset parameters
train_dataset_path: !!str "/volume/ECG_tokenizer/output/MHI/mimic_mhi_psa_train_updated.parquet"
validation_dataset_path: !!str "/volume/ECG_tokenizer/output/MHI/mhi_psa_test_updated.parquet"
num_workers: !!int 32
batch_size: !!int 12
signal_path_column: !!str "ecg_path" # waveform_path_psa or waveform_path_original
# waveform parameters
waveform_length: !!int 2500
num_leads: !!int 12
normalize_waveforms: !!bool False
lead_stats:
  I:
    mean: !!float -0.982577
    std: !!float 33.551235
  II:
    mean: !!float -0.683182
    std: !!float 35.616466
  III:
    mean: !!float 0.299394
    std: !!float 40.365378
  aVR:
    mean: !!float 0.832880
    std: !!float 28.102812
  aVL:
    mean: !!float -0.640986
    std: !!float 32.563651
  aVF:
    mean: !!float -0.191894
    std: !!float 34.169092
  V1:
    mean: !!float -1.351383
    std: !!float 54.328896
  V2:
    mean: !!float -1.366718
    std: !!float 72.522918
  V3:
    mean: !!float -1.325758
    std: !!float 66.522440
  V4:
    mean: !!float -1.091183
    std: !!float 61.623552
  V5:
    mean: !!float -0.937534
    std: !!float 57.341500
  V6:
    mean: !!float -0.882349
    std: !!float 52.247728