Upload weight-space autoencoder (encoder + decoder) and configuration

README.md

@@ -35,8 +35,8 @@ It includes both an encoder (compresses weights into latent representations) and
 
 ## Performance Metrics (Test Set)
 
-- **MSE**: 5.611373
-- **MAE**: 0.956549
-- **RMSE**: 2.368834
-- **Cosine Similarity**: 0.3278
-- **R² Score**: 0.0936
+- **MSE**: 0.188579
+- **MAE**: 0.228755
+- **RMSE**: 0.434257
+- **Cosine Similarity**: 0.9519
+- **R² Score**: 0.9698

config.yaml

@@ -93,7 +93,7 @@ loss:
   reconstruction_type: mse
   temperature: 0.1
   type: contrastive
-run_dir: /Users/max/Desktop/muat/model_zoo/runs/train-encoder-decoder_config_2025-12-14_20-11-52
+run_dir: /Users/max/Desktop/muat/model_zoo/runs/train-encoder-decoder_config_2025-12-14_21-59-52
 run_log_cleanup: false
 tokenization:
   chunk_size: 1

decoder.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2bf7b5742fdcb2e76007c89763c71dc983e68cbc6892336e54183d2c336cb9a1
-size 102545358
+oid sha256:aade98a47efa45b7711844ee501b59b4cc34aaa4345292c7ef6150a06d2f594d
+size 102545422

encoder.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:87d40c3774ad424ae3b178530e1619e8ac02c03e427e7adf684fc40bc6c0d02d
-size 77277100
+oid sha256:d528e825396e5cfd0e806d26a4a603a58aa30c874f8f5d81a00e9144d943ca2c
+size 77285292

tokenizer_config.json

@@ -3,6 +3,7 @@
   "max_tokens": 512,
   "include_metadata": true,
   "metadata_features": 5,
-  "token_dim": 12,
-  "granularity": "neuron"
+  "token_dim": 14,
+  "granularity": "neuron",
+  "max_neuron_data_size": 9
 }