| [20:01:21] - INFO - absl - A polynomial schedule was set with a non-positive `transition_steps` value; this results in a constant schedule with value `init_value`. | |
| Traceback (most recent call last): | |
| File "./run_mlm_flax.py", line 688, in <module> | |
| optax.clip_grad_by_global_norm(1.), | |
| AttributeError: module 'optax' has no attribute 'clip_grad_by_global_norm' |