git.iko.soy/grenade
on master help sign in
Fix off-by-one error in C code
2c68ddcErik de Castro Lopo
cbits/gradient_decent.c+1−1
compare with master
unifiedsplit
@@ -6,7 +6,7 @@ void decend_cpu(int len, double rate, double momentum, double regulariser,
6 const double* last,
6 const double* last,
7 double* outputWeights, double* outputMomentum) {
7 double* outputWeights, double* outputMomentum) {
8
8
9− for (int i = 0; i <= len; i++) {
9+ for (int i = 0; i < len; i++) {
10 outputMomentum[i] = momentum * last[i] - rate * gradient[i];
10 outputMomentum[i] = momentum * last[i] - rate * gradient[i];
11 outputWeights[i] = weights[i] + outputMomentum[i] - (rate * regulariser) * weights[i];
11 outputWeights[i] = weights[i] + outputMomentum[i] - (rate * regulariser) * weights[i];
12 }
12 }