seizuretransformer / config.json
eugenehp's picture
Add SeizureTransformer research-only weights and safety notice
73a6e72 verified
{
"model_type": "SeizureTransformer",
"task": "eeg-seizure-detection",
"in_channels": 19,
"in_samples": 15360,
"dim_feedforward": 2048,
"num_layers": 8,
"num_heads": 4,
"drop_rate": 0.1,
"max_pos_len": 6000,
"output": "time_step_probability",
"output_shape": ["batch", "time"]
}