signal-jepa / config.json
PierreGtch's picture
Signal-JEPA pre-training weights (16s, run adeuwv4s, with channel embeddings)
9e20126 verified
{
"n_outputs": null,
"n_chans": null,
"chs_info": [
{
"ch_name": "Fp1",
"loc": [
-0.0294367,
0.0839171,
-0.00699
]
},
{
"ch_name": "Fp2",
"loc": [
0.0298723,
0.0848959,
-0.00708
]
},
{
"ch_name": "F7",
"loc": [
-0.0702629,
0.0424743,
-0.01142
]
},
{
"ch_name": "F3",
"loc": [
-0.0502438,
0.0531112,
0.042192
]
},
{
"ch_name": "Fz",
"loc": [
0.0003122,
0.058512,
0.066462
]
},
{
"ch_name": "F4",
"loc": [
0.0518362,
0.0543048,
0.040814
]
},
{
"ch_name": "F8",
"loc": [
0.0730431,
0.0444217,
-0.012
]
},
{
"ch_name": "FC5",
"loc": [
-0.0772149,
0.0186433,
0.02446
]
},
{
"ch_name": "FC1",
"loc": [
-0.0340619,
0.0260111,
0.079987
]
},
{
"ch_name": "FC2",
"loc": [
0.0347841,
0.0264379,
0.078808
]
},
{
"ch_name": "FC6",
"loc": [
0.0795341,
0.0199357,
0.024438
]
},
{
"ch_name": "T7",
"loc": [
-0.0841611,
-0.0160187,
-0.009346
]
},
{
"ch_name": "C3",
"loc": [
-0.0653581,
-0.0116317,
0.064358
]
},
{
"ch_name": "Cz",
"loc": [
0.0004009,
-0.009167,
0.100244
]
},
{
"ch_name": "C4",
"loc": [
0.0671179,
-0.0109003,
0.06358
]
},
{
"ch_name": "T8",
"loc": [
0.0850799,
-0.0150203,
-0.00949
]
},
{
"ch_name": "TP9",
"loc": [
-0.0856192,
-0.0465147,
-0.045707
]
},
{
"ch_name": "CP5",
"loc": [
-0.0795922,
-0.0465507,
0.030949
]
},
{
"ch_name": "CP1",
"loc": [
-0.0355131,
-0.0472919,
0.091315
]
},
{
"ch_name": "CP2",
"loc": [
0.0383838,
-0.0470731,
0.090695
]
},
{
"ch_name": "CP6",
"loc": [
0.0833218,
-0.0461013,
0.031206
]
},
{
"ch_name": "TP10",
"loc": [
0.0861618,
-0.0470353,
-0.045869
]
},
{
"ch_name": "P7",
"loc": [
-0.0724343,
-0.0734527,
-0.002487
]
},
{
"ch_name": "P3",
"loc": [
-0.0530073,
-0.0787878,
0.05594
]
},
{
"ch_name": "Pz",
"loc": [
0.0003247,
-0.081115,
0.082615
]
},
{
"ch_name": "P4",
"loc": [
0.0556667,
-0.0785602,
0.056561
]
},
{
"ch_name": "P8",
"loc": [
0.0730557,
-0.0730683,
-0.00254
]
},
{
"ch_name": "PO9",
"loc": [
-0.0549104,
-0.0980448,
-0.035465
]
},
{
"ch_name": "O1",
"loc": [
-0.0294134,
-0.112449,
0.008839
]
},
{
"ch_name": "Oz",
"loc": [
0.0001076,
-0.114892,
0.014657
]
},
{
"ch_name": "O2",
"loc": [
0.0298426,
-0.112156,
0.0088
]
},
{
"ch_name": "PO10",
"loc": [
0.0549876,
-0.0980911,
-0.035541
]
},
{
"ch_name": "FC3",
"loc": [
-0.0601819,
0.0227162,
0.055544
]
},
{
"ch_name": "FC4",
"loc": [
0.0622931,
0.0237228,
0.05563
]
},
{
"ch_name": "C5",
"loc": [
-0.0802801,
-0.0137597,
0.02916
]
},
{
"ch_name": "C1",
"loc": [
-0.036158,
-0.0099839,
0.089752
]
},
{
"ch_name": "C2",
"loc": [
0.037672,
-0.0096241,
0.088412
]
},
{
"ch_name": "C6",
"loc": [
0.0834559,
-0.0127763,
0.029208
]
},
{
"ch_name": "CP3",
"loc": [
-0.0635562,
-0.0470088,
0.065624
]
},
{
"ch_name": "CPz",
"loc": [
0.0003858,
-0.047318,
0.099432
]
},
{
"ch_name": "CP4",
"loc": [
0.0666118,
-0.0466372,
0.06558
]
},
{
"ch_name": "P1",
"loc": [
-0.0286203,
-0.0805249,
0.075436
]
},
{
"ch_name": "P2",
"loc": [
0.0319197,
-0.0804871,
0.076716
]
},
{
"ch_name": "POz",
"loc": [
0.0002156,
-0.102178,
0.050608
]
},
{
"ch_name": "FT9",
"loc": [
-0.0840759,
0.0145673,
-0.050429
]
},
{
"ch_name": "FTT9h",
"loc": [
-0.084125,
-0.0018467,
-0.029794
]
},
{
"ch_name": "TTP7h",
"loc": [
-0.0855651,
-0.0306287,
0.011153
]
},
{
"ch_name": "TP7",
"loc": [
-0.0848302,
-0.0460217,
-0.007056
]
},
{
"ch_name": "TPP9h",
"loc": [
-0.0781602,
-0.0607567,
-0.023824
]
},
{
"ch_name": "FT10",
"loc": [
0.0841131,
0.0143647,
-0.050538
]
},
{
"ch_name": "FTT10h",
"loc": [
0.084123,
-0.0018083,
-0.029638
]
},
{
"ch_name": "TPP8h",
"loc": [
0.0785198,
-0.0604323,
0.012902
]
},
{
"ch_name": "TP8",
"loc": [
0.0855488,
-0.0455453,
-0.00713
]
},
{
"ch_name": "TPP10h",
"loc": [
0.0789027,
-0.0609553,
-0.023805
]
},
{
"ch_name": "F9",
"loc": [
-0.0701019,
0.0416523,
-0.049952
]
},
{
"ch_name": "F10",
"loc": [
0.0721141,
0.0420667,
-0.050452
]
},
{
"ch_name": "AF7",
"loc": [
-0.0548397,
0.0685722,
-0.01059
]
},
{
"ch_name": "AF3",
"loc": [
-0.0337007,
0.0768371,
0.021227
]
},
{
"ch_name": "AF4",
"loc": [
0.0357123,
0.0777259,
0.021956
]
},
{
"ch_name": "AF8",
"loc": [
0.0557433,
0.0696568,
-0.010755
]
},
{
"ch_name": "PO3",
"loc": [
-0.0365114,
-0.1008529,
0.037167
]
},
{
"ch_name": "PO4",
"loc": [
0.0367816,
-0.1008491,
0.036397
]
}
],
"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,
"channel_embedding": "pretrain_aligned",
"braindecode_version": "1.5.0dev0"
}