I\'m working to a classification problem (Emotion Recognition throuhg EEG data) with a custom dataset (not publically available, for now) in pytorch. EEG data a