File size: 2,395 Bytes
c0d66fc ea3803a 2af844a 68cf5e4 228daf2 c0d66fc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 | # 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
|