SleepFMStager / config.json
adammnrr's picture
Mirror the SleepFM sleep stager in braindecode format
8681fba verified
Raw
History Blame Contribute Delete
346 Bytes
{
"n_outputs": 5,
"n_chans": 4,
"chs_info": null,
"n_times": 3840,
"input_window_seconds": null,
"sfreq": 128.0,
"patch_size": 640,
"embed_dim": 128,
"staging_num_heads": 4,
"staging_num_layers": 1,
"staging_pooling_heads": 4,
"drop_prob": 0.3,
"max_seq_length": 8196,
"activation": "torch.nn.modules.activation.ELU"
}