Upload TFRobertaForMaskedLM

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: distilbert/distilroberta-base
+base_model: TheophileCA/distilroberta_trainedOnCACatalogue
 tags:
 - generated_from_keras_callback
 model-index:
@@ -13,10 +13,10 @@ probably proofread and complete it, then remove this comment. -->
 
 # distilroberta_trainedOnCACatalogue
 
-This model is a fine-tuned version of [distilbert/distilroberta-base](https://huggingface.co/distilbert/distilroberta-base) on an unknown dataset.
+This model is a fine-tuned version of [TheophileCA/distilroberta_trainedOnCACatalogue](https://huggingface.co/TheophileCA/distilroberta_trainedOnCACatalogue) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 2.0395
-- Validation Loss: 2.0452
+- Train Loss: 1.8254
+- Validation Loss: 1.7345
 - Epoch: 2
 
 ## Model description
@@ -43,9 +43,9 @@ The following hyperparameters were used during training:
 
 | Train Loss | Validation Loss | Epoch |
 |:----------:|:---------------:|:-----:|
-| 2.2110     | 2.0461          | 0     |
-| 2.1291     | 2.0448          | 1     |
-| 2.0395     | 2.0452          | 2     |
+| 2.2309     | 1.9275          | 0     |
+| 1.9894     | 1.7416          | 1     |
+| 1.8254     | 1.7345          | 2     |
 
 
 ### Framework versions

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "distilbert/distilroberta-base",
+  "_name_or_path": "TheophileCA/distilroberta_trainedOnCACatalogue",
   "architectures": [
     "RobertaForMaskedLM"
   ],

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1cbc19b2af20f5e4c57e3170aff8b7bafc7efdc52220f9539218375890aa4f8c
+oid sha256:2280666d2b4301bc38d5493138af72ef904450d1ba55803db25d270bb86866e5
 size 484838604