-
Notifications
You must be signed in to change notification settings - Fork 44
Description
Hi.
When I'm training yolov4-tiny and setting my executionEnvironment setting is 'gpu', Matlab will print:
Error using gpuArray/subsasgn
When assigning into a GPUArray, the subscripts must contain unique values.Subscript 2 contained repeated values.Error in dlarray/parenAssign (line 53)
objdata(argsNonDlarray{:}) = rhsdata;Error in train>modelGradients (line 229)
tcls_(idx,tcls+1) = 1.0;% 确保类别标签是从0开始标注的索引,否则这里会超出维度Error in deep.internal.dlfeval (line 18)
[varargout{1:nout}] = fun(x{:});Error in dlfeval (line 41)
[varargout{1:nout}] = deep.internal.dlfeval(fun,varargin{:});Error in train (line 121)
[gradients,boxLoss,objLoss,clsLoss,totalLoss,state] =
dlfeval(@modelGradients, model, XTrain, YTrain,yoloLayerNumber);
But if I change executionEnvironment to 'cpu' ,it can successfully do training.
Can you give me some clues about what happened?
Thank for help.