flax.optim 引入错误

flax 在0.6.0 删除了optim 包,
需要下载 0.5.3 版本
详情见:
https://github.com/google/flax/commit/cda7a4c85bbce744e412ab82e298ddf76d4770d2

pip install flax==0.5.3

你可能感兴趣的:(血泪史,python)