{ | |
"whisper_model": "tiny", | |
"sae_model": "pretrained/topkautoencoder_baseline/checkpoints/bestval.pth", | |
"layer_name": "encoder.blocks.2", | |
"batch_size": 160, | |
"data_path": "audio_data/librispeech/LibriSpeech/test-other", | |
"device": "cuda", | |
"out_folder": "activations/pretrained_topk_baseline", | |
"dl_max_workers": 0, | |
"collect_max": null | |
} | |