noisy net

gzz1996 · Dec 18, 2018 · 62bca23 · 62bca23
1 parent 48c04b3
commit 62bca23
Show file tree

Hide file tree

Showing 5 changed files with 119 additions and 94 deletions.
diff --git a/.idea/misc.xml b/.idea/misc.xml
diff --git a/.idea/tensorflow1.2.iml b/.idea/tensorflow1.2.iml
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/RL/Basic-NoisyNet-Demo/utils.py b/RL/Basic-NoisyNet-Demo/utils.py
@@ -22,13 +22,13 @@ def f(e_list):
         inputs = tf.contrib.layers.flatten(inputs)
     flatten_shape = inputs.shape[1]
     weights = tf.get_variable('weights', shape=[flatten_shape, units], initializer=w_i)
-    w_noise = tf.get_variable('w_noise', [flatten_shape, units], initializer=w_i, collections=c_names)
+    w_sigma = tf.get_variable('w_sigma', [flatten_shape, units], initializer=w_i, collections=c_names)
     if noisy_distribution == 'independent':
-        weights += tf.multiply(tf.random_normal(shape=w_noise.shape), w_noise)
+        weights += tf.multiply(tf.random_normal(shape=w_sigma.shape), w_sigma)
     elif noisy_distribution == 'factorised':
         noise_1 = f(tf.random_normal(tf.TensorShape([flatten_shape, 1]), dtype=tf.float32))  # 注意是列向量形式，方便矩阵乘法
         noise_2 = f(tf.random_normal(tf.TensorShape([1, units]), dtype=tf.float32))
-        weights += tf.multiply(noise_1 * noise_2, w_noise)
+        weights += tf.multiply(noise_1 * noise_2, w_sigma)
     dense = tf.matmul(inputs, weights)
     if bias_shape is not None:
         assert bias_shape[0] == units