[19:48:19] - INFO - absl - A polynomial schedule was set with a non-positive `transition_steps` value; this results in a constant schedule with value `init_value`.
/home/dat/pino/lib/python3.8/site-packages/jax/lib/xla_bridge.py:386: UserWarning: jax.host_count has been renamed to jax.process_count. This alias will eventually be removed; please update your code.
  warnings.warn(
/home/dat/pino/lib/python3.8/site-packages/jax/lib/xla_bridge.py:373: UserWarning: jax.host_id has been renamed to jax.process_index. This alias will eventually be removed; please update your code.
  warnings.warn(
Epoch ... (1/3):   0%|                                                                                                                                 | 0/3 [00:00<?, ?it/s]







































































































































Training...:  41%|████████████████████████████████████████████████▋                                                                       | 507/1250 [06:34<10:39,  1.16it/s]







































































































































Training...:  80%|███████████████████████████████████████████████████████████████████████████████████████████████▎                       | 1001/1250 [11:21<25:59,  6.26s/it]