voiceSHIELD-small / training_config.json
sumitranjan's picture
Upload 10 files
8beda60 verified
raw
history blame contribute delete
600 Bytes
{
"mappings_dir": "/content/drive/MyDrive/voice_dataset/mappings",
"output_dir": "/content/whisper-security-model-full",
"drive_backup": "/content/drive/MyDrive/voice_dataset/model_output",
"model_name": "openai/whisper-small",
"num_batches": 17,
"max_duration": 25,
"train_ratio": 0.7,
"val_ratio": 0.15,
"test_ratio": 0.15,
"seed": 42,
"n_folds": 5,
"batch_size": 4,
"grad_accum": 8,
"learning_rate": 3e-05,
"warmup_steps": 200,
"max_steps": 3000,
"logging_steps": 50,
"eval_steps": 200,
"save_steps": 500,
"labels": {
"safe": 0,
"malicious": 1
}
}