incorrect matrix size while training data
1 次查看(过去 30 天)
显示 更早的评论
I am training a neural network with input [25x1x1].
I am taking input using imageInputLayer([25 1]).
My training data is sotred in the variable images which is of size [25x1x1x80000].
But when I run the program to train the network I get an error:
I dont know what is causing this error. My input dimensions match with the dimensions of my training data and everything else in the network seems fine.
Please help.
回答(1 个)
Raynier Suresh
2021-2-17
Hi, check whether you have defined the architecture of your network correctly, you can do this by using the command "analyzeNetwork(layers)". For input layer size [25 1] and the training data size [25 1 1 80000] the trainNetwork function should work fine for example you can refer the code below.
layers = [imageInputLayer([25 1])
convolution2dLayer(1,32)
reluLayer
fullyConnectedLayer(4)
softmaxLayer
classificationLayer];
options = trainingOptions('sgdm', ...
'MaxEpochs',1,...
'InitialLearnRate',1e-4, ...
'Verbose',false, ...
'Plots','training-progress');
Xtrain = rand(25,1,1,80000);
Ytrain = categorical(randi(4,80000,1));
net = trainNetwork(Xtrain,Ytrain,layers,options);
0 个评论
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!