MATLAB Answers

Error using trainNetwork (line 183) Invalid validation data. The output size (10) of the last layer does not match the number of classes (1).

2 views (last 30 days)
Steven Chou
Steven Chou on 10 Jun 2021
Answered: Anshika Chaurasia on 16 Jun 2021
layers = [
imageInputLayer([28 28 1]) %¹Ï¼h¿é¤Jpixel RGB or Grayscale
convolution2dLayer(3,8,'Padding','same') %²Ä¤@¼hconvolution pooling
batchNormalizationLayer
reluLayer
maxPooling2dLayer(2,'Stride',2)
convolution2dLayer(3,16,'Padding','same') %²Ä¤G¼hconvolution pooling
batchNormalizationLayer
reluLayer
maxPooling2dLayer(2,'Stride',2)
convolution2dLayer(3,32,'Padding','same')
batchNormalizationLayer
reluLayer
fullyConnectedLayer(10)
softmaxLayer
classificationLayer];

Answers (1)

Anshika Chaurasia
Anshika Chaurasia on 16 Jun 2021
Hi,
This error can occur because the number of classes defined in the neural network layers is not equal to the number of classes (categories) of the validation data.
I would suggest you to match the outputSize of fullyConnectedLayer with the number of classes of the dataset.
Hope it helps!

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!