How can I use the example Transfer Learning Using Alexnet with Vgg16?
2 Ansichten (letzte 30 Tage)
Ältere Kommentare anzeigen
Claudio Mor
am 12 Nov. 2018
Beantwortet: xu lu
am 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 Kommentare
Akzeptierte Antwort
Johannes Bergstrom
am 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
Weitere Antworten (1)
xu lu
am 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 Kommentare
Siehe auch
Kategorien
Mehr zu Image Data Workflows finden Sie in Help Center und File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!