flax
e3f579a5
- [flax] Add module path to nn.module.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
[flax] Add module path to nn.module. 1. Rename the original scope.path as scope.debug_path. 2. Add a new scope.path that reflects module path. 3. Expose module.scope.path as module.path. PiperOrigin-RevId: 563281823
References
#3300 - [flax] Add module path to nn.module.
Author
JXRiver
Committer
a-googler
Parents
82dd366e
Loading