PierreGtch's picture
Signal-JEPA pre-training weights (16s, run adeuwv4s, without channel embeddings)
48f4634 verified
raw
history blame contribute delete
816 Bytes
{
"n_outputs": null,
"n_chans": null,
"n_times": null,
"input_window_seconds": null,
"sfreq": 128.0,
"feature_encoder__conv_layers_spec": [
[
8,
32,
8
],
[
16,
2,
2
],
[
32,
2,
2
],
[
64,
2,
2
],
[
64,
2,
2
]
],
"drop_prob": 0.0,
"feature_encoder__mode": "default",
"feature_encoder__conv_bias": false,
"activation": "torch.nn.modules.activation.GELU",
"pos_encoder__spat_dim": 30,
"pos_encoder__time_dim": 34,
"pos_encoder__sfreq_features": 1.0,
"pos_encoder__spat_kwargs": null,
"transformer__d_model": 64,
"transformer__num_encoder_layers": 8,
"transformer__num_decoder_layers": 4,
"transformer__nhead": 8,
"braindecode_version": "1.5.0dev0"
}