Skip to content

Commit be50da0

Browse files
Update src/functor.jl
Co-authored-by: Brian Chen <ToucheSir@users.noreply.github.com>
1 parent 4d56e75 commit be50da0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/functor.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ end
5050

5151
"""
5252
params(model)
53+
params(layers...)
5354
5455
Given a model or specific layers from a model, create a `Params` object pointing to its trainable parameters.
5556

0 commit comments

Comments
 (0)