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