error using nnet.inter​nal.cnn.la​yer.util.f​ullyConnec​tedGPUImag​eStrategy

13 ビュー (過去 30 日間)
红婷 郝
红婷 郝 2020 年 7 月 2 日
disp('Track: network training section begins!')
trainOpt = trainingOptions('adam', ...
'InitialLearnRate',options.learningRate, ...
'LearnRateSchedule','piecewise', ...
'LearnRateDropFactor',options.dropFactor, ...
'LearnRateDropPeriod',options.learnRateSch, ...
'L2Regularization',options.weightDecay, ...
'MaxEpochs',options.maxNumEpochs, ...
'MiniBatchSize',options.batchSize, ...
'Shuffle','every-epoch', ...
'ValidationData',dataPrep.val, ...
'ValidationFrequency',options.valFreq, ...
'ExecutionEnvironment','gpu', ...
'ValidationPatience', 10, ... % Disables automatic training break-off
'Plots','none');
gpuDevice(1)
[trainedNet, trainingInfo] = trainNetwork(dataPrep.train{1,:}, dataPrep.train{2,:}, net, trainOpt);
save([options.netSaveDir,'net_ant=',num2str(options.ch(i)),'_pilot=',num2str(options.pilotSize)], 'trainedNet')
% show NMSE on validation dataset
nanLoc = isnan(trainingInfo.ValidationLoss);
valNMSE = trainingInfo.ValidationLoss(~nanLoc);
options.valNMSE = valNMSE;
R_NMSE(j, i) = options.valNMSE(end);

回答 (0 件)

カテゴリ

Help Center および File ExchangeDeep Learning Toolbox についてさらに検索

タグ

Community Treasure Hunt

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

Start Hunting!

Translated by