How can I use the example Transfer Learning Using Alexnet with Vgg16?
4 次查看(过去 30 天)
显示 更早的评论
I tried to use 'Transfer Learning Using AlexNet' with Vgg16 but it failed to start the Training iterations. How can I use this example with Vgg16?
[netTransfer, info] = trainNetwork(augimdsTrain,layers,options);
Error using trainNetwork (line 150) GPU out of memory. Try reducing 'MiniBatchSize' using the trainingOptions function.
*Error in TL_CM_V3_Test_IM_VGG16 (line 68) [netTransfer, info] = trainNetwork(augimdsTrain,layers,options);
Caused by: Error using .* Out of memory on device. To view more detail about available memory on the GPU, use 'gpuDevice()'. If the problem persists, reset the GPU by calling 'gpuDevice(1)'.*
My gpu is CUDADevice with properties:
Name: 'GeForce 930MX'
Index: 1
ComputeCapability: '5.0'
0 个评论
采纳的回答
Johannes Bergstrom
2018-11-12
Vgg16 requires a lot of GPU memory and you don't have very much of it. The error message says "Try reducing 'MiniBatchSize' using the trainingOptions function." Did you try that? Otherwise, I would recommend using a network that uses less memory, for example, GoogLeNet or SqueezeNet.
You can use any pretriained network available in MATLAB for transfer learning in this example: https://www.mathworks.com/help/deeplearning/examples/train-deep-learning-network-to-classify-new-images.html
For a list of pretrained networks, see https://www.mathworks.com/help/deeplearning/ug/pretrained-convolutional-neural-networks.html
更多回答(1 个)
xu lu
2019-1-4
I have tried many times but failed to install Vgg16 successfully. The installation always stops when the download reaches 18%.Can you help me?
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!