Trixy3/models/wakeword/wakeword_500ep/training_config.json

{
  "trainer_name": "wakeword_trainer",
  "model_name": "wakeword_500ep",
  "output_dir": "./models/wakeword",
  "data_dir": "./trainer/data/wakeword",
  "config_dir": "./config",
  "batch_size": 32,
  "learning_rate": 0.001,
  "num_epochs": 500,
  "min_epochs": 50,
  "max_epochs": 1000,
  "early_stopping_patience": 15,
  "model_format": ".pth",
  "use_mixed_precision": true,
  "gradient_clip_norm": 1.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": 1.5,
  "n_mels": 40,
  "n_fft": 512,
  "hop_length": 160,
  "win_length": 400,
  "use_augmentation": false,
  "noise_factor": 0.1,
  "speed_factor": 0.1,
  "pitch_factor": 0.05,
  "volume_factor": 0.2,
  "use_password_protection": true,
  "password": null,
  "encryption_algorithm": "AES256",
  "log_level": "INFO",
  "save_checkpoints": true,
  "checkpoint_interval": 10,
  "validate_interval": 1,
  "log_interval": 100,
  "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": {
    "model_type": "standard",
    "wakeword_classes": {
      "0": "custom",
      "1": "system_command",
      "2": "negative"
    },
    "detection_threshold": 0.5,
    "use_focal_loss": true,
    "focal_alpha": 1.0,
    "focal_gamma": 2.0,
    "width_multiplier": 1.0,
    "use_se": false,
    "dropout_rate": 0.2
  }
}