Model save

README.md

@@ -1,8 +1,8 @@
 ---
 license: apache-2.0
+base_model: distilroberta-base
 tags:
 - generated_from_trainer
-base_model: distilroberta-base
 model-index:
 - name: gennlp_hw1_reranker
   results: []
@@ -39,11 +39,11 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 3
+- num_epochs: 5
 
 ### Framework versions
 
-- Transformers 4.37.2
+- Transformers 4.38.1
 - Pytorch 2.1.0+cu121
-- Datasets 2.17.1
+- Datasets 2.18.0
 - Tokenizers 0.15.2

config.json

@@ -21,7 +21,7 @@
   "position_embedding_type": "absolute",
   "problem_type": "single_label_classification",
   "torch_dtype": "float32",
-  "transformers_version": "4.37.2",
+  "transformers_version": "4.38.1",
   "type_vocab_size": 1,
   "use_cache": true,
   "vocab_size": 50265

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9f2ad0ac54358220636ff2679b0febf68161cc07c57f12d22cdc5ee34f40db2c
+oid sha256:02b65b12c27570e9660ec3c8986b5eb38ed018fb9ed54f66232b46fc4ffe5eea
 size 328492280

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ba11ef970fe5b23acde680593dd3b6f53914de290cd9acbe9b0076b914b63c97
-size 4728
+oid sha256:06799f90b9378db5396c0e698bb046a5889bb3473636397a50f9b68397ddaacb
+size 4920