AttributeError:模块‘亚麻’没有属性'nn‘

Question

我正在尝试运行RegNeRF，这需要亚麻。在安装最新版本的flax==0.6.0时，我发现了一个错误，说明flax没有属性optim。这个答案建议将亚麻降至0.5.1。在这样做的时候，我得到了错误AttributeError: module 'flax' has no attribute 'nn'

对于此错误，我在网上找不到任何解决方案。任何帮助都是非常感谢的。

我用的是ubuntu 20.04

Answer 1

从Flex0.6.0版本开始，flax.optim模块已被移动到optax；有关如何迁移代码的信息，请参阅将我的代码库升级为Optax。如果使用的是导入flax.optim的外部代码，并且无法更新这些引用，则必须安装FASXVersion0.5.3或更高版本。

关于flax.nn：在亚麻0.4.0版中，这个模块被flax.linen所取代。有关此迁移的信息，请参阅将我的代码库升级到亚麻。如果您使用的是导入flax.nn的外部代码，并且无法更新这些引用，则必须安装FLASXVersion0.3.6或更高版本。