ecg-lead-generator / config.json
rishsoraganvi's picture
Add model weights, config, and architecture
637ec8e verified
{
"model_type": "ecg-lead-generator",
"architecture": "CLIP-Conditioned 1D U-Net",
"n_in": 7,
"n_out": 5,
"base_ch": 64,
"clip_dim": 1024,
"clip_model": "openai/clip-vit-large-patch14",
"dropout": 0.1,
"seq_len": 2500,
"fs_hz": 500,
"input_leads": [
"I",
"II",
"III",
"aVR",
"aVL",
"aVF",
"V1"
],
"output_leads": [
"V2",
"V3",
"V4",
"V5",
"V6"
],
"framework": "pytorch",
"total_parameters": 13703507,
"loss": "huber",
"dataset": "PTB-XL (PhysioNet)",
"training_hardware": "Vast.ai A100"
}