We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f9d9bd0 commit 4c8c816Copy full SHA for 4c8c816
src/trainable.jl
@@ -23,9 +23,9 @@ Flux.trainable(k::RationalQuadraticKernel) = (k.α,)
23
24
Flux.trainable(κ::KernelProduct) = k.kernels
25
26
-Flux.trainable(κ::KernelSum) = (k.weights, k.kernels) #To check
+Flux.trainable(κ::KernelSum) = (κ.weights, κ.kernels) #To check
27
28
-Flux.trainable(k::ScaledKernel) = (k.σ, k.kernel)
+Flux.trainable(κ::ScaledKernel) = (κ.σ, κ.kernel)
29
30
Flux.trainable(κ::TransformedKernel) = (κ.transform, κ.kernel)
31
0 commit comments