bruAristimunha's picture
Add converted EEGSym Keras weights
1bd2ac7 verified
Raw
History Blame Contribute Delete
930 Bytes
{
"architecture": "replicability.eegsym_faithful.FaithfulEEGSym",
"canonical_channel_order": [
"F3",
"C3",
"P3",
"Cz",
"Pz",
"F4",
"C4",
"P4"
],
"checkpoint_format": "torch_state_dict",
"commit": "a76ee9bf66204b4ebd07746aeb48933dd4272daa",
"h5_loader_diagnostics": {
"consumed": 146,
"filled": 146,
"h5_datasets": 146,
"model_params_to_fill": 146
},
"input_shape": [
"batch",
8,
384
],
"keras_h5_sha256": "f60ddd220cf48dd18dd9706c5389c70a244fe878610021c2ed97d40105145b2b",
"keras_h5_source_path": "research/evidence/source_code_repos/EEGSym/EEGSym/EEGSym_pretrained_weights_8_electrode.h5",
"n_chans": 8,
"n_outputs": 2,
"n_times": 384,
"probe_logits_shape": [
2,
2
],
"probe_logits_sum": -0.044638995081186295,
"sample_rate_hz": 128,
"state_count_excluding_num_batches_tracked": 144440,
"trainable_param_count": 142784
}