Add Flux 0.13 compatibility (#103)

FluxML · Mar 9, 2022 · 6a20051 · 6a20051
1 parent ea8071c
commit 6a20051
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -29,7 +29,7 @@ Animations = "0.4"
 BSON = "0.2, 0.3"
 DataStructures = "0.18"
 EarlyStopping = "0.1, 0.2, 0.3"
-Flux = "0.11, 0.12"
+Flux = "0.11, 0.12, 0.13"
 Glob = "1.3"
 Graphs = "1"
 ImageCore = "0.8, 0.9"

diff --git a/src/learner.jl b/src/learner.jl
@@ -137,7 +137,7 @@ function _dataiters(t::Tuple)
 end
 
 
-paramsrec(m) = params(m)
+paramsrec(m) = Flux.params(m)
 paramsrec(t::Union{Tuple,NamedTuple}) = map(paramsrec, t)
 
 # Callback utilities