Runtime error on DEAP Dataset

[ale-pavel]

Hi,

I've tried to run the Sal_model.py on DEAP, but apart from fixing the number of output units of the classifier (using 4 for the valence/arousal quadrants) I can't get it to work because of an error:

RuntimeError: shape '[168, 5, 1, -1]' is invalid for input of size 43352064
The size is the product of 8064x32x168 (at least that's what it seems), of time samples x channels x (batch_size=128 + 40?)

The line is the following:
https://github.com/numediart/Emotion-EEG/blob/master/Models_DEAP.py#L530

Also I implemented the feature generation part myself, as it was incomplete (only the sensor locations code conversion is in the repo), by using "mne.time_frequency.psd_array_welch" and then averaging the power values of 5 frequency bins that are mentioned in the DEAP paper (for a total of 5 image channels which should be correct).

Best Regards.