Update VGCL.py

yimutianyang · Jun 2, 2023 · 377a867 · 377a867
1 parent 403bd31
commit 377a867
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/models/VGCL.py b/models/VGCL.py
@@ -36,12 +36,12 @@ def graph_encoder(self):
             all_emb.append(ego_emb)
         mean = tf.reduce_mean(all_emb, axis=0)
         logstd = tf.matmul(mean, self.eps_weight) + self.eps_bias
-        std = tf.exp(logstd) * 0.01
+        std = tf.exp(logstd)
         ### reparameterization
         noise1 = tf.random_normal(std.shape)
         noise2 = tf.random_normal(std.shape)
-        noised_emb1 = mean + std * noise1
-        noised_emb2 = mean + std * noise2
+        noised_emb1 = mean + 0.01 * std * noise1
+        noised_emb2 = mean + 0.01 * std * noise2
         return noised_emb1, noised_emb2, mean, std