| /home/dat/pino/lib/python3.8/site-packages/jax/lib/xla_bridge.py:382: 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:369: 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/5): 0%| | 0/5 [00:00<?, ?it/s] | |
| Training...: 1%| | 500/92767 [06:10<14:45:27, 1.74it/s] | |
| Training...: 1%|▏ | 1009/92767 [11:19<11:54:13, 2.14it/s] | |