heartwise
/

DeepECG-Tok_EfficientNetV2_77_Classes

Model card Files Files and versions

robana commited on Jan 28

Commit

228daf2

·

verified ·

1 Parent(s): 68cf5e4

Update base_config.yaml

Files changed (1) hide show

base_config.yaml +2 -2

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 "/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"

 num_epochs: !!int 10
 seed: !!int 42
 base_checkpoint_path: !!str "/volume/ECG_tokenizer/outputs/"
+pretrained_tokenizer_path: !!str "/app/checkpoints//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"