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
Show older comments
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.
Answers (1)
ammar noori
on 22 Nov 2021
0 votes
same issue i got. did you fixed this problem?
Categories
Find more on Deep Learning Toolbox in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!