Upload 6 files
Browse files- checkpoints/bestval.pth +3 -0
- events.out.tfevents.1731284424.ozen.358964.0 +3 -0
- events.out.tfevents.1731296736.ozen.513498.0 +3 -0
- features.json +11 -0
- mo.bestval +0 -0
- tiny_l1.json +30 -0
checkpoints/bestval.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:27c34219e7649548ec530445f4141f5165007b7cac51997572e0fc4efc887f6f
|
3 |
+
size 928128
|
events.out.tfevents.1731284424.ozen.358964.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1d3046de84d8e154fc8b7976b16bcc7cccc42e154773244bc8de27516d036bb1
|
3 |
+
size 1989423
|
events.out.tfevents.1731296736.ozen.513498.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:aed6d93cc01be2fffdf6eb8d507617cb30e618a00345b4643b2a2f30250f77d9
|
3 |
+
size 5520701
|
features.json
ADDED
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"whisper_model": "tiny",
|
3 |
+
"sae_model": "pretrained/l1autoencoder_baseline/checkpoints/bestval.pth",
|
4 |
+
"layer_name": "encoder.blocks.2",
|
5 |
+
"batch_size": 160,
|
6 |
+
"data_path": "audio_data/librispeech/LibriSpeech/test-other",
|
7 |
+
"device": "cuda",
|
8 |
+
"out_folder": "activations/pretrained_l1_baseline",
|
9 |
+
"dl_max_workers": 0,
|
10 |
+
"collect_max": null
|
11 |
+
}
|
mo.bestval
ADDED
Binary file (311 kB). View file
|
|
tiny_l1.json
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"whisper_config": {
|
3 |
+
"model": "tiny",
|
4 |
+
"layer_name": "encoder.blocks.2"
|
5 |
+
},
|
6 |
+
"autoencoder_variant": "l1",
|
7 |
+
"autoencoder_config": {
|
8 |
+
"n_dict_components": 200,
|
9 |
+
"recon_alpha": 1e4
|
10 |
+
},
|
11 |
+
"seed": 0,
|
12 |
+
"train_folder": "activations/tiny_block_2_train",
|
13 |
+
"val_folder": "activations/tiny_block_2_dev",
|
14 |
+
"device": "cuda",
|
15 |
+
"run_dir": "runs/l1autoencoder_baseline",
|
16 |
+
"lr": 4e-4,
|
17 |
+
"weight_decay": 0.0,
|
18 |
+
"steps": 100000,
|
19 |
+
"clip_thresh": 1.0,
|
20 |
+
"batch_size": 100,
|
21 |
+
"dl_max_workers": 0,
|
22 |
+
"log_tb_every": 10,
|
23 |
+
"save_every": 1000,
|
24 |
+
"val_every": 100,
|
25 |
+
"optimizer": "radam",
|
26 |
+
"scheduler": "cosine",
|
27 |
+
"scheduler_params": {},
|
28 |
+
"start_checkpoint": null,
|
29 |
+
"from_disk": true
|
30 |
+
}
|