correct wrong fan_in (previously taken from one layer too late)

marph91 · May 13, 2021 · 1cd81c1 · 1cd81c1
1 parent 616961d
commit 1cd81c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/playground/04_custom_toplevel.py b/playground/04_custom_toplevel.py
@@ -646,7 +646,7 @@ def bnn_from_larq(path: str) -> Bnn:
  # used at the next batch norm
  fan_in = (
  get_kernel_size(parameter["kernel_size"]) ** 2
- * parameter["filters"]
+ * channel
  * channel_bw
  )
  # used at the next conv