/home/nipunsadvilkar/roberta_mr_env/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/nipunsadvilkar/roberta_mr_env/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/8): 0%| | 0/8 [00:00<?, ?it/s] | |
Epoch ... (1/8): 12%|███████████████████ | 1/8 [01:15<08:49, 75.59s/it] | |
Epoch ... (1/8): 25%|██████████████████████████████████████ | 2/8 [01:20<03:23, 33.95s/it] | |
Epoch ... (1/8): 38%|█████████████████████████████████████████████████████████ | 3/8 [01:25<01:43, 20.77s/it] | |
Training...: 51%|█████████████████████████████████████████████████████████████████████████████▌ | 72/142 [00:02<00:01, 35.13it/s] | |
Step... (500 | Loss: 8.018753051757812, Learning Rate: 0.0001500000071246177) | |
Step... (500 | Loss: 8.205772399902344, Acc: 0.0773010179400444): 38%|██████████████████████████████████████▋ | 3/8 [01:33<02:35, 31.10s/it] |