Format code

Author: adrhill

src/lrp/canonize.jl

@@ -6,7 +6,7 @@ function fuse_batchnorm(d::Dense, bn::BatchNorm)
     b = if d.bias != false
         scale .* (d.bias - bn.μ) + bn.β
     else
-        - scale .* bn.μ + bn.β
+        -scale .* bn.μ + bn.β
     end
     return Dense(W, b, bn.λ)
 end
@@ -18,7 +18,7 @@ function fuse_batchnorm(c::Conv, bn::BatchNorm)
     b = if c.bias != false
         scale .* (c.bias - bn.μ) + bn.β
     else
-        - scale .* bn.μ + bn.β
+        -scale .* bn.μ + bn.β
     end
     return Conv(bn.λ, W, b, c.stride, c.pad, c.dilation, c.groups)
 end

test/test_canonize.jl

@@ -57,7 +57,7 @@ model = Chain(
     BatchNorm(2, softplus),
     BatchNorm(2),
     flatten,
-    Dense(72, 10, bias=false),
+    Dense(72, 10; bias=false),
     BatchNorm(10),
     BatchNorm(10),
     BatchNorm(10, relu),
@@ -113,7 +113,7 @@ model = Chain(
         +,
         Conv((3, 3), 4 => 5, identity),
         Chain(
-            Conv((3, 3), 4 => 5, bias=false, identity), # fuse
+            Conv((3, 3), 4 => 5; bias=false, identity), # fuse
             BatchNorm(5),
         ),
         Chain(