Error in training SegNet on CPU
5 views (last 30 days)
Show older comments
I am trying to train SegNet on a CPU with an Intel Core i7-4770 processor @ 3.40 GHz using the example code provided ( https://www.mathworks.com/help/vision/examples/semantic-segmentation-using-deep-learning.html ), but I am getting the following error:
Error using trainNetwork (line 154)
Too many input arguments.
Error in exampleSegnet (line 98)
[net, info] = trainNetwork(pximds,lgraph,options);
Caused by:
Error using gather
Too many input arguments.
I know the gather function collects arrays into memory, so could this error be caused by the computer running out of memory? I have about 25 free GB of memory. Additionally, the error persisted even when I reduced the proportion of training images to 10%.
Thanks
0 Comments
Accepted Answer
Vishal Bhutani
on 31 Aug 2018
Based on my understanding you want to train a Semantic Segmentation using CPU with an Intel Core i7-4770 processor. I also think that it may be an issue with the memory. As you maybe aware, that it is mentioned in the documentation, model was trained using NVIDIA Titan X, which has 12GB of GPU memory. As mentioned in the documentation did you try lowering the MiniBatchSize property in training options to 1? If it still not works you may still need to reduce the training data. It might be also possible that the example require GPU for training.
options = trainingOptions('sgdm', ...
'Momentum',0.9, ...
'InitialLearnRate',1e-3, ...
'L2Regularization',0.0005, ...
'MaxEpochs',100, ...
'MiniBatchSize',4, ...
'CheckpointPath',tempdir, ...
'Shuffle','every-epoch', ...
'VerboseFrequency',2);
5 Comments
Walter Roberson
on 23 Nov 2019
evianita dewi asks Jordan Houri:
I have the same error. How did you solve the problem ?
More Answers (0)
See Also
Categories
Find more on GPU Computing in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!