heartwise
/

DeepECG-Tok_EfficientNetV2_77_Classes

Model card Files Files and versions

robana commited on Jan 28

Commit

68cf5e4

·

1 Parent(s): c0d66fc

updated config file

Files changed (1) hide show

base_config.yaml +3 -3

base_config.yaml CHANGED Viewed

@@ -6,9 +6,9 @@ run_mode: !!str "inference"
 num_epochs: !!int 10
 seed: !!int 42
 base_checkpoint_path: !!str "/volume/ECG_tokenizer/outputs/"
-pretrained_tokenizer_path: !!str "/volume/ECG_tokenizer/checkpoints/ECG_tokenizer_latest/best_model_epoch_10.pt"
-inference_dataset_path: !!str "/volume/ECG_tokenizer/outputs/preprocessed_bert_output.parquet"
-inference_checkpoint_path: !!str "/volume/ECG_tokenizer/checkpoints/ECG_Tokenizer_Linear_Probing/ECG_Tokenizer_Linear_Probing/5zg01bx6_20250824-041452/checkpoint_epoch_10.pt"
 # wandb parameters
 wandb_project: !!str "ECG_Tokenizer_Linear_Probing"
 wandb_entity: !!str "mhi_ai"

 num_epochs: !!int 10
 seed: !!int 42
 base_checkpoint_path: !!str "/volume/ECG_tokenizer/outputs/"
+pretrained_tokenizer_path: !!str "/app/checkpoints/deepecg_tokenizer.pt"
+inference_dataset_path: !!str "/app/outputs/preprocessed_bert_output.parquet"
+inference_checkpoint_path: !!str "/app/checkpoints/deepecg_tokenizer_efficientnet.pt"
 # wandb parameters
 wandb_project: !!str "ECG_Tokenizer_Linear_Probing"
 wandb_entity: !!str "mhi_ai"