EMG2Pose-emg2pose / config.json
bruAristimunha's picture
Rehost regression_emg2pose.ckpt with braindecode parameter names
9032d95 verified
Raw History Blame Contribute Delete
777 Bytes
{
"n_outputs": 20,
"n_chans": 16,
"chs_info": null,
"n_times": 11790,
"input_window_seconds": null,
"sfreq": 2000.0,
"encoder_channels": 256,
"feature_dim": 64,
"hidden_size": 512,
"lstm_layers": 2,
"decoder": "mlp",
"decoder_hidden_sizes": [
512,
512
],
"stem_kernel_sizes": [
11,
5
],
"stem_strides": [
5,
2
],
"tds_subsample_kernels": [
17,
9
],
"tds_subsample_strides": [
4,
2
],
"tds_kernel_widths": [
9,
5
],
"tds_blocks": 2,
"tds_channels": 16,
"rollout_rate": 50.0,
"num_position_steps": 500,
"parameterization": "position",
"output_scalar": 0.01,
"activation": "torch.nn.modules.activation.LeakyReLU",
"drop_prob": 0.0,
"braindecode_version": "1.8.0dev0"
}