From 9361292c429c0ba3400c31c7fa5d5e3d3cb6ab47 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 19 Jul 2016 21:50:01 +0000
Subject: [PATCH] updates
---
src/connected_layer.c | 4 +++-
1 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/src/connected_layer.c b/src/connected_layer.c
index f20aa93..e7784ea 100644
--- a/src/connected_layer.c
+++ b/src/connected_layer.c
@@ -192,6 +192,9 @@
l.weights[i*l.inputs + j] *= scale;
}
l.biases[i] -= l.rolling_mean[i] * scale;
+ l.scales[i] = 1;
+ l.rolling_mean[i] = 0;
+ l.rolling_variance[i] = 1;
}
}
@@ -257,7 +260,6 @@
axpy_ongpu(l.outputs, 1, l.biases_gpu, 1, l.output_gpu + i*l.outputs, 1);
}
activate_array_ongpu(l.output_gpu, l.outputs*l.batch, l.activation);
-
}
void backward_connected_layer_gpu(connected_layer l, network_state state)
--
Gitblit v1.10.0