From 735867872deb22c3731a9b4b5b6fb774fe7897d4 Mon Sep 17 00:00:00 2001 From: Marcus Chiam Date: Fri, 26 Jan 2024 15:40:55 -0800 Subject: [PATCH] updated nowrap docstring --- flax/linen/module.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/flax/linen/module.py b/flax/linen/module.py index 8fb4497d9c..31c955ac54 100644 --- a/flax/linen/module.py +++ b/flax/linen/module.py @@ -508,7 +508,10 @@ def nowrap(fun: _CallableT) -> _CallableT: - if you're subclassing a method like Module.param and don't want this overriden core function decorated with the state management wrapper. - If you want a method to be callable from an unbound Module (e.g.: a - function of construction of arguments that doesn't depend on params/RNGs) + function of construction of arguments that doesn't depend on params/RNGs). + If you want to learn more about how Flax Modules manage their state read the + [The Flax Module lifecycle](https://flax.readthedocs.io/en/latest/developer_notes/module_lifecycle.html) + guide. For instance::