Disable bias before batch norm layer

RubixML · Jun 18, 2020 · 005e219 · 005e219
1 parent d4a4ce8
commit 005e219
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -97,7 +97,7 @@ $estimator = new PersistentModel(
         new Activation(new LeakyReLU()),
         new Dense(100),
         new Activation(new LeakyReLU()),
-        new Dense(100),
+        new Dense(100, 0.0, false),
         new BatchNorm(),
         new Activation(new LeakyReLU()),
         new Dense(50),

diff --git a/train.php b/train.php
@@ -49,7 +49,7 @@
         new Activation(new LeakyReLU()),
         new Dense(100),
         new Activation(new LeakyReLU()),
-        new Dense(100),
+        new Dense(100, 0.0, false),
         new BatchNorm(),
         new Activation(new LeakyReLU()),
         new Dense(50),