Error in training SegNet on CPU

Question

Jordan Houri 2018-8-23

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/416035-error-in-training-segnet-on-cpu

编辑： MShia 2019-11-26

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 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Vishal Bhutani 2018-8-31

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/416035-error-in-training-segnet-on-cpu#answer_334835

在 MATLAB Online 中打开

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);