| [13:56:50] - WARNING - __main__ - Unable to display metrics through TensorBoard because some package are not installed: No module named 'tensorflow' | |
| Epoch ... (1/3): 0%| | 0/3 [00:00<?, ?it/s] | |
| Training...: 2%|████▏ | 1000/54217 [18:07<14:47:08, 1.00s/it] | |
| Training...: 2%|████▏ | 1000/54217 [18:09<14:47:08, 1.00s/it] | |
| Evaluating ...: 0%| | 0/1167 [00:00<?, ?it/s] | |
| Training...: 4%|████████▎ | 2000/54217 [46:46<14:17:16, 1.02it/s] | |
| Training...: 4%|████████▎ | 2000/54217 [46:47<14:17:16, 1.02it/s] | |
| Step... (2000 | Loss: 1.330615520477295, Acc: 0.7090045213699341): 0%| | 0/3 [1:15:55<?, ?it/s] | |
| Traceback (most recent call last): | |
| File "run_mlm_flax.py", line 815, in <module> | |
| main() | |
| File "run_mlm_flax.py", line 718, in main | |
| samples = [tokenized_datasets["train"][int(idx)] for idx in batch_idx] | |
| File "run_mlm_flax.py", line 718, in <listcomp> | |
| samples = [tokenized_datasets["train"][int(idx)] for idx in batch_idx] | |
| File "/data/flax/lib/python3.8/site-packages/jax/_src/device_array.py", line 247, in <genexpr> | |
| return (sl for chunk in self._chunk_iter(100) for sl in chunk._unstack()) | |
| File "/data/flax/lib/python3.8/site-packages/jax/_src/numpy/lax_numpy.py", line 6792, in _chunk_iter | |
| for i in range(num_chunks): | |
| File "/usr/lib/python3.8/functools.py", line 399, in _method | |
| return self.func(cls_or_self, *self.args, *args, **keywords) | |
| File "/data/flax/lib/python3.8/site-packages/jax/_src/device_array.py", line 41, in _forward_method | |
| return fun(getattr(self, attrname), *args) | |
| KeyboardInterrupt |