/usr/lib/python3/dist-packages/requests/__init__.py:89: RequestsDependencyWarning: urllib3 (1.26.9) or chardet (3.0.4) doesn't match a supported version!
  warnings.warn("urllib3 ({}) or chardet ({}) doesn't match a supported "
Reusing dataset imdb (/home/ashvid/.cache/huggingface/datasets/imdb/plain_text/1.0.0/2fdd8b9bcadd6e7055e742a706876ba43f19faee861df134affd7a3f60fc38a1)
100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 3/3 [00:03<00:00,  1.11s/it]
Parameter 'function'=<function tokenize_function at 0x7fa3c3014430> of the transform datasets.arrow_dataset.Dataset._map_single couldn't be hashed properly, a random hash was used instead. Make sure your transforms and parameters are serializable with pickle or dill for the dataset fingerprinting and caching to work. If you reuse this transform, the caching mechanism will consider it to be different from the previous calls and recompute everything. This warning is only showed once. Subsequent hashing failures won't be showed.
100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 25/25 [00:03<00:00,  6.59ba/s]
100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 25/25 [00:03<00:00,  6.91ba/s]
100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 50/50 [00:07<00:00,  6.90ba/s]
2022-07-04 13:45:47.475528: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  SSE3 SSE4.1 SSE4.2 AVX AVX2 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2022-07-04 13:45:47.476829: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1532] Created device /job:localhost/replica:0/task:0/device:GPU:0 with 30204 MB memory:  -> device: 0, name: AMD Radeon PRO W6800, pci bus id: 0000:03:00.0
Epoch 1/5
2022-07-04 13:45:57.202048: I tensorflow/core/common_runtime/gpu_fusion_pass.cc:507] ROCm Fusion is enabled.
125/125 [==============================] - ETA: 0s - loss: 0.7263 - sparse_categorical_accuracy: 0.51702022-07-04 13:46:43.151218: I tensorflow/core/common_runtime/gpu_fusion_pass.cc:507] ROCm Fusion is enabled.
125/125 [==============================] - 69s 471ms/step - loss: 0.7263 - sparse_categorical_accuracy: 0.5170 - val_loss: 0.7354 - val_sparse_categorical_accuracy: 0.5120
Epoch 2/5
125/125 [==============================] - 58s 466ms/step - loss: 0.7125 - sparse_categorical_accuracy: 0.4980 - val_loss: 0.7159 - val_sparse_categorical_accuracy: 0.4880
Epoch 3/5
125/125 [==============================] - 58s 469ms/step - loss: 0.7052 - sparse_categorical_accuracy: 0.5110 - val_loss: 0.7384 - val_sparse_categorical_accuracy: 0.4880
Epoch 4/5
125/125 [==============================] - 59s 469ms/step - loss: 0.7114 - sparse_categorical_accuracy: 0.5160 - val_loss: 0.6955 - val_sparse_categorical_accuracy: 0.4880
Epoch 5/5
125/125 [==============================] - 59s 471ms/step - loss: 0.7117 - sparse_categorical_accuracy: 0.4620 - val_loss: 0.7205 - val_sparse_categorical_accuracy: 0.4880