Removed trainmode from tests
This commit is contained in:
parent
5cbd2cecf2
commit
568ecb1c97
|
@ -2,7 +2,6 @@ using Flux, Test, Statistics
|
||||||
using Zygote: pullback
|
using Zygote: pullback
|
||||||
|
|
||||||
evalwgrad(f, x...) = pullback(f, x...)[1]
|
evalwgrad(f, x...) = pullback(f, x...)[1]
|
||||||
trainmode(f) = (testmode!(f, false); f)
|
|
||||||
|
|
||||||
@testset "Dropout" begin
|
@testset "Dropout" begin
|
||||||
x = [1.,2.,3.]
|
x = [1.,2.,3.]
|
||||||
|
|
Loading…
Reference in New Issue