Skip to content

Commit 5357883

Browse files
Copilotyebai
andcommitted
Fix Literate.jl parsing issue in deep kernel learning example
Co-authored-by: yebai <[email protected]>
1 parent 910e73e commit 5357883

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/2-deep-kernel-learning/script.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,9 @@ rng = Random.default_rng()
5454
ps, st = Lux.setup(rng, neuralnet)
5555

5656
# Create a wrapper function for the neural network that will be updated during training
57-
function neural_transform(x, params)
58-
# neuralnet returns (output, new_state), we only need the output
59-
return first(neuralnet(x, params, st))
57+
function neural_transform(x, θ)
58+
# neuralnet returns (output, new_state), we only need the output
59+
return first(neuralnet(x, θ, st))
6060
end
6161

6262
# We use the Squared Exponential Kernel:

0 commit comments

Comments
 (0)