taco/microWakeWord-Trainer-Nvidia-Docker
1
0
Fork 0
mirror of https://github.com/TaterTotterson/microWakeWord-Trainer-Nvidia-Docker.git synced 2026-06-12 20:10:19 -06:00
Code Issues Packages Projects Releases Wiki Activity

Update advanced_training_notebook.ipynb

Browse Source
This commit is contained in:
MasterPhooey
2025-01-05 16:47:10 -06:00
committed by GitHub
parent cc962f2069
commit 8d1289ed49
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
advanced_training_notebook.ipynb
View File

@@ -527,7 +527,7 @@
"config[\"features\"] = [\n", "config[\"features\"] = [\n",
" {\n", " {\n",
" \"features_dir\": \"generated_augmented_features\",\n", " \"features_dir\": \"generated_augmented_features\",\n",
" \"sampling_weight\": 5.0, # Increased\n", " \"sampling_weight\": 2.0, # Increased\n",
" \"penalty_weight\": 1.0,\n", " \"penalty_weight\": 1.0,\n",
" \"truth\": True,\n", " \"truth\": True,\n",
" \"truncation_strategy\": \"truncate_start\",\n", " \"truncation_strategy\": \"truncate_start\",\n",
@@ -535,7 +535,7 @@
" },\n", " },\n",
" {\n", " {\n",
" \"features_dir\": \"negative_datasets/speech\",\n", " \"features_dir\": \"negative_datasets/speech\",\n",
" \"sampling_weight\": 8.0, # Adjusted\n", " \"sampling_weight\": 10.0, # Adjusted\n",
" \"penalty_weight\": 1.0,\n", " \"penalty_weight\": 1.0,\n",
" \"truth\": False,\n", " \"truth\": False,\n",
" \"truncation_strategy\": \"random\",\n", " \"truncation_strategy\": \"random\",\n",
@@ -543,7 +543,7 @@
" },\n", " },\n",
" {\n", " {\n",
" \"features_dir\": \"negative_datasets/dinner_party\",\n", " \"features_dir\": \"negative_datasets/dinner_party\",\n",
" \"sampling_weight\": 8.0, # Adjusted\n", " \"sampling_weight\": 10.0, # Adjusted\n",
" \"penalty_weight\": 1.0,\n", " \"penalty_weight\": 1.0,\n",
" \"truth\": False,\n", " \"truth\": False,\n",
" \"truncation_strategy\": \"random\",\n", " \"truncation_strategy\": \"random\",\n",
@@ -569,20 +569,20 @@
"\n", "\n",
"config[\"training_steps\"] = [20000] # Increased\n", "config[\"training_steps\"] = [20000] # Increased\n",
"config[\"positive_class_weight\"] = [1]\n", "config[\"positive_class_weight\"] = [1]\n",
"config[\"negative_class_weight\"] = [15] # Adjusted\n", "config[\"negative_class_weight\"] = [20] # Adjusted\n",
"config[\"learning_rates\"] = [0.0005] # Adjusted\n", "config[\"learning_rates\"] = [0.001] # Adjusted\n",
"config[\"batch_size\"] = 128\n", "config[\"batch_size\"] = 128\n",
"\n", "\n",
"config[\"time_mask_max_size\"] = [30] # Enabled SpecAugment\n", "config[\"time_mask_max_size\"] = [0] # Enabled SpecAugment\n",
"config[\"time_mask_count\"] = [2]\n", "config[\"time_mask_count\"] = [0]\n",
"config[\"freq_mask_max_size\"] = [15]\n", "config[\"freq_mask_max_size\"] = [0]\n",
"config[\"freq_mask_count\"] = [2]\n", "config[\"freq_mask_count\"] = [0]\n",
"\n", "\n",
"config[\"eval_step_interval\"] = 1000 # Adjusted\n", "config[\"eval_step_interval\"] = 500 # Adjusted\n",
"config[\"clip_duration_ms\"] = 2000 # Increased\n", "config[\"clip_duration_ms\"] = 1500 # Increased\n",
"\n", "\n",
"config[\"target_minimization\"] = 0.9\n", "config[\"target_minimization\"] = 0.9\n",
"config[\"minimization_metric\"] = \"false_positive_rate\" # Updated\n", "config[\"minimization_metric\"] = None # Updated\n",
"config[\"maximization_metric\"] = \"average_viable_recall\"\n", "config[\"maximization_metric\"] = \"average_viable_recall\"\n",
"\n", "\n",
"with open(os.path.join(\"training_parameters.yaml\"), \"w\") as file:\n", "with open(os.path.join(\"training_parameters.yaml\"), \"w\") as file:\n",