Skip to content

Commit 98d8e4f

Browse files
committed
modify scripts for merl & siglip
1 parent b7ce9d9 commit 98d8e4f

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

ecg_bench/scripts/train_1st.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#!/bin/bash
22

33
# models=("stmem" "merl" "mlae" "mtae" "siglip" "clip" "vit")
4-
models=("siglip")
4+
models=("merl")
55
# data=("ecg-qa-mimic-iv-ecg-250-1250")
66
# data=("ecg_instruct_45k_mapped_1250")
77

ecg_bench/scripts/train_2nd.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,9 @@ encoders_checkpoints=("merl_adam_64_50_1024_0.0001_0.9_0.99_1e-08_500_0.01_True_
66
# llms=("gemma-2-2b-it" "llama-3.2-1b-instruct" "qwen2.5-1.5b-instruct")
77
llms=("llama-3.2-1b-instruct")
88
# datasets=("ecg-qa_ptbxl-250-1250" "ecg-qa-mimic-iv-ecg-250-1250" "ecg-instruct-45k-250-1250" "ecg-instruct-pulse-250-1250" "pretrain-mimic-250-1250") # add more datasets here
9-
datasets=("ecg_instruct_45k_mapped_1250")
9+
# datasets=("ecg_instruct_45k_mapped_1250")
10+
datasets=("ecg-qa_mimic-iv-ecg_mapped_1250")
11+
# datasets=("ecg-qa_mimic-iv-ecg_mapped_1250" "ecg-qa_ptbxl_mapped_1250")
1012
# ----------------------------------------------------------
1113

1214
for data in "${datasets[@]}"; do

0 commit comments

Comments
 (0)