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 a5e5546 commit 0a765b1Copy full SHA for 0a765b1
test/train.jl
@@ -46,6 +46,8 @@ end
46
end
47
48
@testset "non-tuple data" begin
49
+ w = randn(10, 10)
50
+ w2 = randn(10, 10)
51
loss(m, x) = Flux.Losses.mse(w*x, m.weight*x .+ m.bias)
52
model = (weight=copy(w2), bias=zeros(10))
53
opt = Flux.setup(AdamW(), model)
0 commit comments