File size: 2,228 Bytes
f291f93 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
/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]
|