AttributeError: module 'flax' has no attribute 'nn'

Nagabhushan S N

I'm trying to run RegNeRF, which requires flax. On installing the latest version of flax==0.6.0, I got an error stating flax has no attribute optim. This answer suggested to downgrade flax to 0.5.1. On doing that, now I'm getting the error AttributeError: module 'flax' has no attribute 'nn'

I could not find any solutions on the web for this error. Any help is appreciated.

I'm using ubuntu 20.04

jakevdp

The flax.optim module has been moved to optax as of flax version 0.6.0; see Upgrading my Codebase to Optax for information on how to migrate your code. If you're using external code that imports flax.optim and can't update these references, you'll have to install flax version 0.5.3 or older.

Regarding flax.nn: this module was replaced by flax.linen in flax version 0.4.0. See Upgrading my Codebase to Linen for information on this migration. If you're using external code that imports flax.nn and can't update these references, you'll have to install flax version 0.3.6 or older.

Collected from the Internet

Please contact [email protected] to delete if infringement.

edited at2022-10-5

Comments

0 comments

TOP Ranking

Article

AttributeError: module 'flax' has no attribute 'nn'

AttributeError: module 'flax' has no attribute 'nn'

Can't pre-populate phone number and message body in SMS link on iPhones when SMS app is not running in the background

pump.io port in URL

Failed to listen on localhost:8000 (reason: Cannot assign requested address)

Spring Boot JPA PostgreSQL Web App - Internal Authentication Error

How to import an asset in swift using Bundle.main.path() in a react-native native module

Loopback Error: connect ECONNREFUSED 127.0.0.1:3306 (MAMP)

BigQuery - concatenate ignoring NULL

Modbus Python Schneider PM5300

Make a B+ Tree concurrent thread safe

What is the exact difference between “ use_all_dns_ips” and "resolve_canonical_bootstrap_servers_only” in client.dns.lookup options?

Svchost high CPU from Microsoft.BingWeather app errors

3D Touch Peek Swipe Like Mail

Printing Int array and String array in one

java.lang.NullPointerException: Cannot read the array length because "<local3>" is null

UWP access denied

Emulator wrong screen resolution in Android Studio 1.3

Google Chrome Translate Page Does Not Work

Apache rewrite or susbstitute rule for bugzilla HTTP 301 redirect

Can a 32-bit antivirus program protect you from 64-bit threats

Is there an option for a Simulink Scope to display the layout in single column?

Unable to use switch toggle for dark mode in material-ui