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

Question

我的代码如下：

!pip install flax
init_params = TransporterNets().init(key, init_img, init_text, init_pix)['params']
print(f'Model parameters: {n_params(init_params):,}')
optim = flax.optim.Adam(lr=1e-4).create(init_params)

但是，它显示了以下错误：

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

尽管我在flax模块中看到了有关flax属性的文档。怎么修呢？

Answer 1

您可以暂时通过将亚麻版本从0.6.0降到0.5.1来解决这个问题。

pip install flax==0.5.1