Error Using trainNetwork in Deep Learning.
显示 更早的评论
Hi,
I am trying to train a pretrained CNN model, inceptionresnetv2 for Image Classification. I am getting the following error,
Error using trainNetwork (line 150)
Error using 'predict' in Layer nnet.inceptionresnetv2.layer.ScalingFactorLayer. The function threw an error and could not be executed.
Any solution??
Thank you!!
6 个评论
Gabija Marsalkaite
2019-5-3
Hi Kaniska,
Could you provide more information about the problem e.g. hardware used (GPU and amount of RAM) and if that is the full error message?
Kaniska Samanta
2019-5-4
编辑:Kaniska Samanta
2019-5-5
Maria Duarte Rosa
2019-5-17
Using a smaller network (e.g. squeeznet or mobilenetv2) and reducing the mini-batch size might help.
Gabija Marsalkaite
2019-5-20
Hi Kaniska,
Most likely your GPU is either low on memory or out of memory. I don't know the exact settings you are using but reducing mini-batch size or using CPU instead of GPU may help (CPU would use 8GB of RAM instead of GPU using 2GB of its memory).
Kaniska Samanta
2019-5-21
Ansuman Mahapatra
2020-7-2
Reduced batch size to 16 and it works.
回答(0 个)
类别
在 帮助中心 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!

