How can I use the example Transfer Learning Using Alexnet with Vgg16?
2 vues (au cours des 30 derniers jours)
Afficher commentaires plus anciens
Claudio Mor
le 12 Nov 2018
Réponse apportée : xu lu
le 4 Jan 2019
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 commentaires
Réponse acceptée
Johannes Bergstrom
le 12 Nov 2018
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
Plus de réponses (1)
xu lu
le 4 Jan 2019
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 commentaires
Voir également
Catégories
En savoir plus sur Image Data Workflows dans Help Center et File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!