Retrain stage2 6-class (F1_macro=0.8127, F1_weighted=0.8357)

stage2/config.json

@@ -7,7 +7,7 @@
     "Spam",
     "Neutral"
   ],
-  "f1_macro": 0.8092020160778541,
+  "f1_macro": 0.8127381267945536,
   "encoder": "microsoft/harrier-oss-v1-0.6b",
   "input_dim": 1024,
   "hidden_dim": 512

stage2/encoder/config_sentence_transformers.json

@@ -12,6 +12,6 @@
   "__version__": {
     "sentence_transformers": "5.3.0",
     "transformers": "5.5.0",
-    "pytorch": "2.11.0+cu130"
+    "pytorch": "2.5.1+cu124"
   }
 }

stage2/encoder/tokenizer_config.json

@@ -5,7 +5,7 @@
   "clean_up_tokenization_spaces": false,
   "eos_token": "<|im_end|>",
   "errors": "replace",
-  "is_local": false,
+  "is_local": true,
   "model_max_length": 131072,
   "pad_token": "<|endoftext|>",
   "split_special_tokens": false,

stage2/neural_weights.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a796a2dd90586a1f41892f95332518bd256b9c79970b677f92825d01c4d1dc5f
-size 2633897
+oid sha256:2b198d5a9a703ce2fdae5df0e457e147d5cd45376bbff1e78c8bb96c8a861d23
+size 2633486