Training completed!

README.md

@@ -1,5 +1,5 @@
 ---
-base_model: vinai/phobert-large
+base_model: vinai/phobert-base
 tags:
 - generated_from_trainer
 model-index:
@@ -12,7 +12,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # CS505-Dev-CSI-PhoBERT_base
 
-This model is a fine-tuned version of [vinai/phobert-large](https://huggingface.co/vinai/phobert-large) on the None dataset.
+This model is a fine-tuned version of [vinai/phobert-base](https://huggingface.co/vinai/phobert-base) on the None dataset.
 
 ## Model description
 
@@ -32,8 +32,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 16
-- eval_batch_size: 8
+- train_batch_size: 32
+- eval_batch_size: 16
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
@@ -46,6 +46,6 @@ The following hyperparameters were used during training:
 ### Framework versions
 
 - Transformers 4.39.3
-- Pytorch 2.2.1+cu121
+- Pytorch 2.1.2
 - Datasets 2.18.0
 - Tokenizers 0.15.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9efac360f120d16c4dd4109f43d55290dcfd87cae4db466c0bd6c8e17fac9b96
+oid sha256:177742eb6e1ae294afe81a261d267e1246330c6f1d2f2c704d323636c4db7a12
 size 540023384

runs/Apr17_09-45-46_b814bffdaab7/events.out.tfevents.1713347148.b814bffdaab7.34.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e5cb2915d252af3f9f4a00e5e96ad8290bf0caac575caceafefa13f780d6f3e0
-size 5772
+oid sha256:b008e2c025d5d893e6e7c08168443d76f6a36b171db4f137ae61514cae50703d
+size 6548