AttributeError: модуль «лен» не имеет атрибута «оптимальный»

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'

Несмотря на то, что я видел документациюoptimатрибут вflaxмодуль. Как это исправить?

Answer 1

Вы можете временно решить проблему, понизив версию льна с 0.6.0 до 0.5.1 на данный момент.

      pip install flax==0.5.1