function trainedNet = EEGNetTraining(data, labels, layers)
   
    options = trainingOptions('sgdm', ...
    'InitialLearnRate',0.001, ...
    'MaxEpochs',100, ...
    'MiniBatchSize',1, ...
    'Shuffle','every-epoch', ...
    'Verbose',false, ...
    'Plots','none');
    
%     data = dlarray(data, 'SSCB');
    data = permute(data, [2 3 4 1]);
    labels = categorical(labels)';
    trainedNet = trainNetwork(data, labels, layers, options);

end