| { | |
| "emb_dims": 384, | |
| "sbert_model_id": "sentence-transformers/all-MiniLM-L12-v2", | |
| "whisper_model_id": "openai/whisper-tiny", | |
| "pooling_mode": "mean", | |
| "with_bidirectionality": false, | |
| "n_new_dims": 10, | |
| "use_psych": true, | |
| "loss": "NCE", | |
| "tau": 0.1, | |
| "batch_size": 1, | |
| "num_workers": 16, | |
| "num_epochs": 50, | |
| "shuffle": false, | |
| "learning_rate": 1e-05, | |
| "weight_decay": 0.01, | |
| "device": "cpu" | |
| } |