change uniform to he_uniform at example

apache · Jun 19, 2020 · 1559b27 · 1559b27
1 parent 7d7d02c
commit 1559b27
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/rbm/train.py b/examples/rbm/train.py
@@ -116,7 +116,7 @@ def train(data_file, use_gpu, num_epoch=10, batch_size=100):
         tvalidposhidprob = tvalidposhidprob + thbias
         tvalidposhidprob = tensor.sigmoid(tvalidposhidprob)
         tvalidposhidrandom = tensor.Tensor(tvalidposhidprob.shape, dev)
-        initializer.uniform(tvalidposhidrandom, 0.0, 1.0)
+        initializer.he_uniform(tvalidposhidrandom, 0.0, 1.0)
         tvalidposhidsample = tensor.gt(tvalidposhidprob, tvalidposhidrandom)
 
         tvalidnegdata = tensor.mult(tvalidposhidsample, tweight.T())