MATLAB Answers

I want to train a network and save the weights. Then use a new data set and resume training with the weights from the previous data set.

2 views (last 30 days)
ronald ogren
ronald ogren on 16 Mar 2018
Commented: ronald ogren on 25 Apr 2018
It seems train always randomizes the weights when called. I can't seem to maintain the weights from the first training set.

  0 Comments

Sign in to comment.

Accepted Answer

Sonam Gupta
Sonam Gupta on 28 Mar 2018
You can continue the training from weights obtained by previous data set by extracting the layers from the network's "Layers" property, and then passing it to "trainNetwork", as follows:
if true
% Train a network
net = trainNetwork(XTrain, YTrain, layers, options);
% Extract layers from the trained network
newLayers = net.Layers;
% Retrain the network, but start from where we left off
newNet = trainNetwork(XTrain, YTrain, newLayers, options);
'trainNetwork' will always use the weights that are stored in the layers which you pass in for training.

More Answers (0)

Sign in to answer this question.