[12:20:22] - WARNING - __main__ - Unable to display metrics through TensorBoard because some package are not installed: No module named 'tensorflow'
/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:18<09:06, 78.12s/it]


Epoch ... (1/8):  25%|██████████████████████████████████████                                                                                                                  | 2/8 [01:22<03:30, 35.03s/it]


Epoch ... (1/8):  38%|█████████████████████████████████████████████████████████                                                                                               | 3/8 [01:27<01:46, 21.23s/it]

Epoch ... (1/8):  38%|█████████████████████████████████████████████████████████                                                                                               | 3/8 [01:31<02:31, 30.34s/it]