Upload 6 files

config.json

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "roberta-base",
   "architectures": [
     "RobertaForRegression"
   ],
@@ -9,17 +10,11 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
-  "id2label": {
-    "0": "LABEL_0"
-  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
-  "label2id": {
-    "LABEL_0": 0
-  },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,
-  "model_type": "roberta",
+  "model_type": "roberta-regression",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pad_token_id": 1,
@@ -28,6 +23,5 @@
   "type_vocab_size": 1,
   "use_cache": true,
   "vocab_size": 50265,
-  "num_labels": 1,
-  "your_custom_parameter": "regression"
+  "output_size": 1
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d2adcf8223a5809360079a0b8a05b15c8527d1fe3f4c13e7f5476dd3f49876f3
-size 498714279
+oid sha256:4cf1a66a1bcd0e64ea3d31d41585095dcff70e26e2d87034a79e7b8a069e42be
+size 498669678