Update src/functor.jl

Co-authored-by: Brian Chen <[email protected]>
FluxML · Nov 27, 2021 · be50da0 · be50da0
1 parent 4d56e75
commit be50da0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/functor.jl b/src/functor.jl
@@ -50,6 +50,7 @@ end
 
 """
     params(model)
+    params(layers...)
 
 Given a model or specific layers from a model, create a `Params` object pointing to its trainable parameters.