kaustavbhattacharjee/finetuning-DistillBERT-amazon-polarity

README.md

@@ -18,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3148
+- Loss: 0.4168
 - Accuracy: 0.89
-- F1: 0.8817
+- F1: 0.8842
 
 ## Model description
 
@@ -39,9 +39,9 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0002
-- train_batch_size: 16
-- eval_batch_size: 16
+- learning_rate: 2e-05
+- train_batch_size: 32
+- eval_batch_size: 32
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5b0e0620c23305d1f1f940af4cdd1b4fcf7ccad62a40f3de617570404477ba77
+oid sha256:91d4e762fa2b8054e9ebf6472681c89ef90b628686ab6d9a88c0e61cc35d7e0b
 size 267832560

runs/Mar06_00-02-34_8f4cc56f19e2/events.out.tfevents.1709683358.8f4cc56f19e2.1462.6

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5844bab870ea3e987f28b5f1b508ff39888f6568604c7c314d1093bdd5352d48
+size 4947

runs/Mar06_00-02-34_8f4cc56f19e2/events.out.tfevents.1709685719.8f4cc56f19e2.1462.7

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc468d4adf4e1334b77720a4d49b745e5b291ef794f3a1db5eaf009d1b27c6ba
+size 812

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37aae881554bbb796b08b8f168ef1f8e266d013e0daaac7b42ff47cdd8e5e7cd
+oid sha256:d282cde220b44b06c79a935e0cecab1d20b03d70bacb5d6afbc6f0b8100c97db
 size 4920