trainer_config.json 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. {
  2. "wakeword": {
  3. "model_name": "custom",
  4. "epochs": 100,
  5. "batch_size": 32,
  6. "learning_rate": 0.001,
  7. "validation_split": 0.2,
  8. "augmentation_enabled": true,
  9. "early_stopping_patience": 10
  10. },
  11. "voice_recognition": {
  12. "model_name": "default",
  13. "epochs": 100,
  14. "batch_size": 16,
  15. "learning_rate": 0.0001,
  16. "validation_split": 0.2,
  17. "embedding_size": 256,
  18. "augmentation_enabled": true,
  19. "early_stopping_patience": 10
  20. },
  21. "audio_processing": {
  22. "sample_rate": 16000,
  23. "n_mels": 80,
  24. "n_fft": 1024,
  25. "hop_length": 256,
  26. "win_length": 1024,
  27. "fmin": 0.0,
  28. "fmax": 8000.0
  29. },
  30. "data": {
  31. "raw_data_directory": "trainer/data",
  32. "processed_data_directory": "trainer/processed",
  33. "cache_directory": "trainer/cache",
  34. "min_samples_per_class": 10
  35. },
  36. "output": {
  37. "models_directory": "models",
  38. "export_onnx": true,
  39. "export_torchscript": true,
  40. "model_password": ""
  41. },
  42. "device": "auto",
  43. "num_workers": 4,
  44. "seed": 42,
  45. "verbose": true
  46. }