Skip to content

Commit 2b7c548

Browse files
committed
Adding jvdp1's review
1 parent 2e64151 commit 2b7c548

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/nf/nf_conv1d_layer_submodule.f90

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ pure module subroutine backward(self, input, gradient)
118118

119119
!--- Compute bias gradients: db(n) = sum_j gdz(n, j)
120120
do n = 1, self % filters
121-
db_local(n) = sum(gdz(n, :))
121+
db_local(n) = sum(gdz(n, :), dim=1)
122122
end do
123123

124124
!--- Initialize weight gradient and input gradient accumulators.

0 commit comments

Comments
 (0)