File size: 606 Bytes
637ec8e | 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 | {
"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"
} |