Trixy3/models/voice_recognition/basic_voice/training_config.json

{
  "trainer_name": "voice_recognition_trainer",
  "model_name": "basic_voice",
  "output_dir": "./models/voice_recognition",
  "data_dir": "./trainer/data/voice_recognition",
  "config_dir": "./config",
  "batch_size": 32,
  "learning_rate": 0.001,
  "num_epochs": 200,
  "min_epochs": 20,
  "max_epochs": 1000,
  "early_stopping_patience": 20,
  "model_format": ".pth",
  "use_mixed_precision": false,
  "gradient_clip_norm": 5.0,
  "weight_decay": 0.0001,
  "validation_split": 0.2,
  "test_split": 0.1,
  "shuffle_data": true,
  "num_workers": 4,
  "pin_memory": true,
  "sample_rate": 16000,
  "bit_depth": 16,
  "audio_length": 3.0,
  "n_mels": 40,
  "n_fft": 512,
  "hop_length": 160,
  "win_length": 400,
  "use_augmentation": false,
  "noise_factor": 0.05,
  "speed_factor": 0.05,
  "pitch_factor": 0.02,
  "volume_factor": 0.1,
  "use_password_protection": true,
  "password": null,
  "encryption_algorithm": "AES256",
  "log_level": "INFO",
  "save_checkpoints": true,
  "checkpoint_interval": 10,
  "validate_interval": 1,
  "log_interval": 100,
  "use_lightweight_validation": true,
  "lightweight_sample_ratio": 0.3,
  "lightweight_max_samples": 1000,
  "comprehensive_validation_epochs": null,
  "device": "cuda",
  "use_distributed": false,
  "num_gpus": 1,
  "resume_from_checkpoint": false,
  "checkpoint_path": null,
  "freeze_backbone": false,
  "use_scheduler": true,
  "scheduler_type": "cosine",
  "custom_params": {
    "mode": "train",
    "debug": false,
    "model_format": ".pth",
    "embedding_dim": 192,
    "use_password_protection": false
  }
}