Skip to content

Commit

Permalink
chore: apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
  • Loading branch information
avik-pal and github-actions[bot] committed Nov 21, 2024
1 parent 3ddd9c5 commit a08903d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions lib/LuxLib/test/normalization/batchnorm_tests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -145,7 +145,7 @@ end
@test nt.running_var isa aType && length(nt.running_var) == 6

@test_gradients(sumabs2first, batchnorm, x, scale, bias, Constant(running_mean),
Constant(running_var), Val(true), gelu, 0.9, 1e-5; atol=1.0f-3, rtol=1.0f-3,
broken_backends=[AutoEnzyme()])
Constant(running_var), Val(true), gelu, 0.9, 1e-5; atol=1.0f-3,
rtol=1.0f-3, broken_backends=[AutoEnzyme()])
end
end

0 comments on commit a08903d

Please sign in to comment.