|
/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 ///google/ |
|
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 ; 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] |
|
|
|
|
|
|
