/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 [07:17<17:29:11, 1.47it/s] | |
Training...: 1%|▌ | 1000/92767 [12:59<17:22:21, 1.47it/s] | |
Training...: 2%|▊ | 1502/92767 [19:04<132:13:12, 5.22s/it] | |