Во время обучения, почему на устройстве больше не существует данных, возникает ошибка. - PullRequest
0 голосов
/ 18 сентября 2018

Ошибка при использовании trainNetwork (строка 140) Данные больше не существуют на устройстве.

Ошибка в ownet2 (строка 50) net = trainNetwork (imdstrain, layer, opts);

Причинаby: Ошибка при использовании nnet.internal.cnngpu.reluForward. Данные больше не существуют на устройстве.

мой код

clc; clear;
imdstrain = imageDatastore('D:\train\elbow','IncludeSubfolders',true,'LabelSource', 'foldernames');
imdstrain.ReadFcn = @(filename)readAndPreproc(filename);
layers = [
    imageInputLayer([256 256 3])
      convolution2dLayer(5,96)
      reluLayer
      convolution2dLayer(5,192)
      reluLayer
      maxPooling2dLayer(2,'Stride',2)
      convolution2dLayer(3,256)
      reluLayer
      convolution2dLayer(3,256)
      reluLayer
      maxPooling2dLayer(2,'Stride',2)
      convolution2dLayer(3,384)
      reluLayer
      convolution2dLayer(3,384)
      reluLayer  
      maxPooling2dLayer(2,'Stride',2)
      convolution2dLayer(3,256)
      reluLayer
      convolution2dLayer(3,256)
      reluLayer
      fullyConnectedLayer(5)
      softmaxLayer
      classificationLayer];
opts = trainingOptions('sgdm', ...
    'InitialLearnRate', 0.001, ... 
    'LearnRateSchedule', 'piecewise', ...
    'L2Regularization', 0.0001, ... 
    'MaxEpochs', 25, ...
    'MiniBatchSize', 50, ...
    'Verbose', true, ...
    'Plots','training-progress');
net = trainNetwork(imdstrain,layers,opts);
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...