diff --git a/docs/src/guide/models/recurrence.md b/docs/src/guide/models/recurrence.md index 44834e9a8a..aa550a04ac 100644 --- a/docs/src/guide/models/recurrence.md +++ b/docs/src/guide/models/recurrence.md @@ -109,6 +109,7 @@ using Optimisers: AdamW function loss(model, x, y) ŷ = model(x) + y = stack(y, dims=2) return Flux.mse(ŷ, y) end