| [20:40:27] - 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/_src/numpy/lax_numpy.py:3132: UserWarning: Explicitly requested dtype <class 'jax._src.numpy.lax_numpy.int64'> requested in zeros is not available, and will be truncated to dtype int32. To enable more dtypes, set the jax_enable_x64 configuration option or the JAX_ENABLE_X64 shell environment variable. See https://github.com/google/jax#current-gotchas for more. | |
| lax._check_user_dtype_supported(dtype, "zeros") | |
| /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][20:40:28] - INFO - __main__ - Skipping to epoch 0 step 0 | |
| Training...: 20%|█████████████████████████▌ | 250/1250 [04:04<12:00, 1.39it/s] | |
| Training...: 40%|███████████████████████████████████████████████████▏ | 500/1250 [07:05<09:01, 1.39it/s] | |
| Training...: 60%|████████████████████████████████████████████████████████████████████████████▊ | 750/1250 [10:06<06:00, 1.39it/s] | |
| Training...: 80%|█████████████████████████████████████████████████████████████████████████████████████████████████████▌ | 1000/1250 [13:06<03:00, 1.39it/s] | |
| Epoch ... (1/3): 33%|██████████████████████████████████████████▋ | 1/3 [16:12<32:24, 972.31s/it] | |
| Training...: 1%|█▏ | 11/1250 [00:03<03:39, 5.65it/s] | |
| Training...: 21%|███████████████████████████▏ | 266/1250 [03:04<03:01, 5.42it/s] | |