google-deepmind / google-deepmind/deepmind-research
ValueError: All `hk.Module`s must be initialized inside an `hk.transform`.
Open
- Dominant language
- Jupyter Notebook
- Stars
- 15.2k
- Forks
- 2.9k
- PR merge metrics
- No merged PRs in 30d
Description
Code:
`from nfnet import NFNet `
`model = NFNet(num_classes=5)`
Error:
ValueError: All `hk.Module`s must be initialized inside an `hk.transform`.
Contributor guide
Assessment
This issue has not been assessed yet.