wandb/run-20210713_104745-1rl2j7or/files/output.log

/home/dat/pino/lib/python3.8/site-packages/jax/_src/numpy/lax_numpy.py:3114: 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: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...:  60%|██████████████████            | 50/83 [01:32<00:23,  1.40it/s]



Epoch ... (1/5):  20%|█████▍                     | 1/5 [02:00<08:02, 120.70s/it]

Training...:  16%|████▋                         | 13/83 [00:07<00:53,  1.32it/s]







Training...:  78%|███████████████████████▍      | 65/83 [00:44<00:24,  1.38s/it]

Epoch ... (1/5):  40%|███████████▏                | 2/5 [03:06<04:25, 88.56s/it]

Training...:  22%|██████▉                         | 18/83 [00:01<00:07,  9.26it/s]








Epoch ... (1/5):  60%|████████████████▊           | 3/5 [04:12<02:36, 78.08s/it]s]
Step... (150 | Loss: 7.8581647872924805, Learning Rate: 2.256410152767785e-05)

Training...:  33%|███████████                       | 27/83 [00:03<00:06,  9.31it/s]







Training...:  93%|███████████████████████████████▌  | 77/83 [00:32<00:04,  1.41it/s]

Epoch ... (1/5):  80%|██████████████████████▍     | 4/5 [05:18<01:13, 73.25s/it]/it]