cherrvak commited on
Commit
efd52ac
·
verified ·
1 Parent(s): 016952a

Upload 6 files

Browse files
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
+ }