We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 01635ec commit 7e8f41cCopy full SHA for 7e8f41c
lib/WeightInitializers/test/initializers_tests.jl
@@ -359,12 +359,9 @@ end
359
v = kaiming_normal(rng, n_in, n_out)
360
σ2 = sqrt(2 / n_out)
361
362
- if (backend == "cuda" || backend == "amdgpu") && rng isa GPUArrays.RNG
363
- @test_broken 0.9σ2 < std(v) < 1.1σ2
364
- else
365
- @test 0.9σ2 < std(v) < 1.1σ2
366
- end
+ @test 0.9σ2 < std(v) < 1.1σ2 broken = rng isa GPUArrays.RNG
367
end
+
368
# Type
369
@test eltype(kaiming_uniform(rng, 3, 4; gain=1.5f0)) == Float32
370
@test eltype(kaiming_normal(rng, 3, 4; gain=1.5f0)) == Float32
0 commit comments