Upload model

config.json

@@ -7,7 +7,7 @@
   "model": {
     "cls": {
       "combined_dim": 128,
-      "combined_input": 1024,
+      "combined_input": 768,
       "n_layers": 2,
       "num_classes": 1,
       "skip_in": [
@@ -21,7 +21,7 @@
         80
       ],
       "tfidf_hidden_dim": 512,
-      "tfidf_input_dim": 8192,
+      "tfidf_input_dim": 8145,
       "tfidf_output_dim": 128,
       "weight_norm": true
     },
@@ -34,19 +34,6 @@
         80
       ],
       "weight_norm": true
-    },
-    "seq": {
-      "freeze": true,
-      "hidden_dim": 256,
-      "input_dim": 768,
-      "lstm_in": 768,
-      "n_layers": 2,
-      "output_dim": 128,
-      "skip_in": [
-        80
-      ],
-      "use_LSTM": false,
-      "weight_norm": true
     }
   },
   "model_type": "headlineclassifier",
@@ -55,7 +42,7 @@
     "anneal_end": 5,
     "batch_size": 32,
     "end_iter": 10,
-    "learning_rate": 0.0005,
+    "learning_rate": 2e-05,
     "learning_rate_alpha": 0.05,
     "save_freq": 5,
     "val_freq": 1,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ff99f937becf8d9a2e97d4ebd993262caefe40bfe284c78f6ff651391d2fb2a6
-size 1081039772
+oid sha256:e762dac2c01a466cbf1a9b683e4941a26ffd72fbbb00687d9002d5031d10ddd9
+size 456666408