SleepFMStager / config.json
bruAristimunha's picture
Self-contained stager export (braindecode PR #1106 @ b0ae1aa8)
95903b5 verified
Raw History Blame Contribute Delete
552 Bytes
{
"n_outputs": 5,
"n_chans": 4,
"chs_info": null,
"n_times": 3840,
"input_window_seconds": null,
"sfreq": 128.0,
"channel_modalities": null,
"patch_size": 640,
"embed_dim": 128,
"encoder_num_heads": 8,
"encoder_num_layers": 6,
"encoder_pooling_heads": 8,
"encoder_drop_prob": 0.0,
"encoder_chunk_patches": 60,
"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",
"braindecode_version": "1.8.1"
}