gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

Fix std calculation using unbiased estimation in sharing stat mode.

Browse Source
This commit is contained in:
cxx
2017-11-07 13:49:33 +08:00
committed by Itai Caspi
parent f47b8092af
commit 84e536d371
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
architectures/tensorflow_components/shared_variables.py
View File

@@ -40,8 +40,9 @@ class SharedRunningStats(object):
name="count", trainable=False)
self._shape = shape
self._mean = tf.to_float(self._sum / self._count)
self._std = tf.sqrt(tf.maximum(tf.to_float(self._sum_squared / self._count) - tf.square(self._mean), 1e-2))
self._mean = self._sum / self._count
self._std = tf.sqrt(tf.maximum((self._sum_squared - self._count*tf.square(self._mean))
/ tf.maximum(self._count-1, 1), epsilon))
self.new_sum = tf.placeholder(shape=self.shape, dtype=tf.float64, name='sum')
self.new_sum_squared = tf.placeholder(shape=self.shape, dtype=tf.float64, name='var')