Skip to content

Commit a562f1d

Browse files
don't require stable rng
1 parent 054dc29 commit a562f1d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/scimlstructures_interface.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ Zygote.gradient(0.1ones(length(SS.canonicalize(SS.Tunable(), p)[1]))) do tunable
8080
end
8181

8282
using OrdinaryDiffEq
83-
using StableRNGs, Lux
83+
using Random, Lux
8484
using ComponentArrays
8585
using SciMLSensitivity
8686
import SciMLStructures as SS
@@ -120,7 +120,7 @@ end
120120
function initialize()
121121
# Defining the neural network
122122
U = Lux.Chain(Lux.Dense(3,30,tanh),Lux.Dense(30,30,tanh),Lux.Dense(30,1))
123-
rng = StableRNG(1111)
123+
rng = Random.GLOBAL_RNG
124124
_para,st = Lux.setup(rng,U)
125125
_para = ComponentArray(_para)
126126
# Setting the parameters

0 commit comments

Comments
 (0)