Fix stddev to match NoisyNet paper

Kaixhin · Feb 19, 2019 · 6c8b281 · 6c8b281
1 parent a476865
commit 6c8b281
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -6,7 +6,7 @@
 
 # Factorised NoisyLinear layer with bias
 class NoisyLinear(nn.Module):
-  def __init__(self, in_features, out_features, std_init=0.4):
+  def __init__(self, in_features, out_features, std_init=0.5):
     super(NoisyLinear, self).__init__()
     self.in_features = in_features
     self.out_features = out_features