Upload single-layer head (lr=5e-05) + config + README

README.md

@@ -0,0 +1,31 @@
+---
+library_name: pytorch
+tags:
+  - classification
+  - linear-head
+  - single
+  - wildjailbreak-linear-head
+license: apache-2.0
+datasets:
+  - CatBarks/wildjailbreak-train-with-llama2-last-hidden-nq-approved
+model-index:
+  - name: linear-head-single-L1-lr5e-05
+    results:
+      - task:
+          type: text-classification
+          name: Multi-class Classification
+        dataset:
+          name: wildjailbreak vectors
+          type: CatBarks/wildjailbreak-train-with-llama2-last-hidden-nq-approved
+        metrics:
+          - type: f1_macro
+            value: 0.0000
+          - type: accuracy
+            value: 0.0000
+---
+# linear-head-single-L1-lr5e-05
+
+Single-layer linear head trained on last-token features.
+- Arch: `single`
+- Layers: `1`
+- LR: `5e-05`

linear-head-single-L1-lr0.0001.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4f6c81bfa81a822febf60cb081907202a2319a82d086ad540bd5ad73d1ad876b
+size 83810

linear-head-single-L1-lr0.0003.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2fb5ea2b0d807f2d8698e6dbfb2ea152dceeb8d46947e62e797e1ad1a9aff7fd
+size 83810

linear-head-single-L1-lr0.0005.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8bdc7a25634f00c6fd2a53fd7d01aff66d4da475e91a1cec622153fee577eeee
+size 83810

linear-head-single-L1-lr1e-05.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5a0e0bfb8ac6efe34ed7ae2a49e476ca14facb52c855a07fb86e0097b5976a19
+size 83804

linear-head-single-L1-lr1e-06.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2157bdd3b0677f0098faff83b46de98bb2dab6753d7a6dfd02eba44709cc5537
+size 83804

linear-head-single-L1-lr1e-07.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dfde7b99bb214c61b3fa60cac804897c1246d6ef70df222fe00bef86d227f117
+size 83804

linear-head-single-L1-lr3e-05.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f8cf08c1ebdaccb2efeb5fb96a037ee7c722e2ca8e0fac9109c3169f990c9b6
+size 83804

linear-head-single-L1-lr3e-07.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:119184df7e96ad738f48beb6d6adf0fd36889c0f70483291cd52aa19b3182ada
+size 83804

linear-head-single-L1-lr5e-05.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eff1e8a1e45072489823134f4e0708714fc8b10f03a99fc0697adebeec4576eb
+size 83804

linear-head-single-L1-lr5e-05.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:920a1c02bd7f19065d9f5074c950a59e634ced99d47e42559dc81519c1645624
+size 82080

linear-head-single-L1-lr8e-05.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e3e32aafb35e5185d1642e1eb6d395f026fc2e12564c36d7282e1c7a43b8478
+size 83804

model_config.json

@@ -0,0 +1,6 @@
+{
+  "in_dim": 5120,
+  "out_dim": 4,
+  "num_layers": 1,
+  "arch": "single"
+}