error using trainNetwork (line 170), Out of memory. Type "help memory" for your options. caused by: Out of memory. Try "help memory" for your options. when training in semantic segmentation using deep learning

1 Ansicht (letzte 30 Tage)
I have problem when training my dataset in semantic segmentation using deep learning.
I have tried to reduce my batchsize. but still error
options = trainingOptions('sgdm', ...
'ExecutionEnvironment','gpu',...
'LearnRateSchedule','piecewise',...
'LearnRateDropPeriod',10,...
'LearnRateDropFactor',0.3,...
'Momentum',0.9, ...
'InitialLearnRate',1e-3, ...
'L2Regularization',0.005, ...
'MaxEpochs',30, ...
'MiniBatchSize',2, ...
'Shuffle','every-epoch', ...
'CheckpointPath','D:\TAEvianita\video\tilesdatase', ...
'VerboseFrequency',2,...
'Plots','training-progress',...
'ValidationPatience', 4);
I tried training the dataset with the following code
doTraining = true;
if doTraining
[net,info] = trainNetwork(pximds,lgraph,options);
save ('D:\TAEvianita\video\tilesdatase\Pretraineddata.mat','net');
disp('NN trained')
else
%load the pre-trained network
data = load('D:\TAEvianita\video\tilesdatase\net_checkpoint__266__2019_12_23__23_44_28.mat')
net = data.net
end
and here is the error I got:
error using trainNetwork (line 170), Out of memory. Type "help memory" for your options. caused by: Out of memory. Try "help memory" for your options.

Antworten (1)

ammar noori
ammar noori am 22 Nov. 2021
same issue i got. did you fixed this problem?

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by