Skip to content

Commit 88dd99a

Browse files
authored
Merge pull request #112 from EnzymeAD/format-main
Format code of branch "main"
2 parents f9e12e7 + 2b32079 commit 88dd99a

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

test/nn.jl

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,10 +69,13 @@ out2 = model(noisy) # first row is prob. of true, second row p(false)
6969
mean((out2[1, :] .> 0.5) .== truth) # accuracy 94% so far!
7070

7171
@testset "conv: groups $groups" for groups in (1, 2, 4)
72-
nn_conv = Conv(randn(Float32, 10, 10, 8 ÷ groups, groups), randn(Float32, groups); groups)
72+
nn_conv = Conv(
73+
randn(Float32, 10, 10, 8 ÷ groups, groups), randn(Float32, groups); groups
74+
)
7375
conv_reactant = Conv(
74-
Reactant.ConcreteRArray(nn_conv.weight), Reactant.ConcreteRArray(nn_conv.bias);
75-
groups
76+
Reactant.ConcreteRArray(nn_conv.weight),
77+
Reactant.ConcreteRArray(nn_conv.bias);
78+
groups,
7679
)
7780

7881
img = randn(Float32, 224, 224, 8, 2)

0 commit comments

Comments
 (0)