Fix fullyconnected error on gpuarray

mldiego · Mar 27, 2024 · 3126ebb · 3126ebb
1 parent 9c7ee10
commit 3126ebb
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/code/nnv/engine/nn/layers/FullyConnectedLayer.m b/code/nnv/engine/nn/layers/FullyConnectedLayer.m
@@ -289,7 +289,6 @@
                     I = reshape(I,N,1); % flatten input
                     if i==1
                         V(1, 1,:,i) = obj.Weights*I + obj.Bias;
-                        V2(1,1,:,i) = gpuArray(obj.Weights)*I + gpuArray(obj.Bias);
                     else
                         V(1, 1,:,i) = obj.Weights*I;
                     end