Skip to content

Commit 813986a

Browse files
committed
Exchange rng in init of RQSCouplingBlocks
1 parent 5b5310b commit 813986a

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/rqspline_coupling.jl

+2-3
Original file line numberDiff line numberDiff line change
@@ -112,9 +112,8 @@ Construct and instance of `RQSplineCouplingBlock`, while initializing the parame
112112
compute device specified in `compute_unit`. (Defaults to CPU)
113113
"""
114114
function RQSplineCouplingBlock(mask::Vector{Bool}, nn::Chain, compute_unit::AbstractComputeUnit=CPUnit())
115-
# Set fixed rng to be consistent across julia versions, alternative: rng = Random.default_rng()
116-
rng = TaskLocalRNG()
117-
Random.seed!(rng, 0)
115+
# Set fixed rng to be consistent across julia versions, alternative: rng = Random.default_rng(); Random.seed!(rng,0)
116+
rng = MersenneTwister(1234)
118117

119118
lux_compute_unit = compute_unit isa CPUnit ? cpu_device() : gpu_device()
120119

0 commit comments

Comments
 (0)