loading and training an existing network.

조회 수: 3 (최근 30일)
Mark Hubelbank
Mark Hubelbank 2024년 10월 7일
편집: Matt J 2024년 10월 8일
I am trying define a network, then train it in multiple sessions. The problem is that I can't get the load or read of the network to work in the second session. The code is:
layers = [ ...
sequenceInputLayer(270)
bilstmLayer(numHiddenUnits,OutputMode="last")
fullyConnectedLayer(numClasses)
softmaxLayer
classificationLayer]
options = trainingOptions("adam", ...
InitialLearnRate=0.002,...
MaxEpochs=15, ...
Shuffle="never", ...
GradientThreshold=1, ...
Verbose=false, ...
ExecutionEnvironment="gpu", ...
Plots="training-progress");
clabels=categorical(labels);
numLables=numel(clabels)
load("savednet.mat","layers");
net = trainNetwork(data,clabels,layers,options);
save("savednet","net");
I have tried many variations of the load command and it always gives an error on the second argument:
Warning: Variable 'layers' not found.
Exactly what should that look like and then how should it be used as input to the trainNetwork routine?
  댓글 수: 7
Mark Hubelbank
Mark Hubelbank 2024년 10월 7일
이동: Walter Roberson 2024년 10월 7일
Perhaps I don't understand how one can train in stages then. The idea is that the training will be continued in the second and subsequent sessions. Sort of a continuing transfer learning. The idea is that over time the network keeps improving. perhaps I should be using trainnet instead of trainnetwork. Then it would appear the call is:
load(filename,"net1","layers");
net=trainnet(data,clabels,net1,"crossentropy",options);
Is this the correct direction?
Walter Roberson
Walter Roberson 2024년 10월 7일
Probably
net1 = trainnet(data,clabels,net1,"crossentropy",options);

댓글을 달려면 로그인하십시오.

채택된 답변

Matt J
Matt J 2024년 10월 7일
previous = load("savednet","net","layers");
net = trainNetwork(data,clabels,previous.net,options);
  댓글 수: 1
Matt J
Matt J 2024년 10월 7일
편집: Matt J 2024년 10월 8일
perhaps I should be using trainnet instead of trainnetwork.
It would be better, since trainnet is newer and has more flexibility. However, it won't make a difference as far as how to resume the training of a pre-existing network..

댓글을 달려면 로그인하십시오.

추가 답변 (0개)

카테고리

Help CenterFile Exchange에서 Image Data Workflows에 대해 자세히 알아보기

제품


릴리스

R2024b

Community Treasure Hunt

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

Start Hunting!

Translated by