| [20:31:32] - 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:31:33] - INFO - __main__ - Skipping to epoch 0 step 0 | |
| Training...: 20%|████████████████████████ | 250/1250 [04:11<11:42, 1.42it/s] | |
| Training...: 40%|████████████████████████████████████████████████ | 500/1250 [07:07<08:46, 1.42it/s] | |