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-04 21:52:10 -06:00
committed by GitHub
parent 17c3c69292
commit c52a9fc45c
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
advanced_training_notebook.ipynb
View File

@@ -673,22 +673,22 @@
"# Training command with optimized settings\n", "# Training command with optimized settings\n",
"!\"{sys.executable}\" -m microwakeword.model_train_eval \\\n", "!\"{sys.executable}\" -m microwakeword.model_train_eval \\\n",
"--training_config='training_parameters.yaml' \\\n", "--training_config='training_parameters.yaml' \\\n",
"--train 1 \\ # Enable training\n", "--train 1 \\\n",
"--restore_checkpoint 1 \\ # Resume training from the last checkpoint\n", "--restore_checkpoint 1 \\\n",
"--test_tf_nonstreaming 1 \\ # Test TensorFlow non-streaming model\n", "--test_tf_nonstreaming 1 \\\n",
"--test_tflite_nonstreaming 1 \\ # Test TFLite non-streaming model\n", "--test_tflite_nonstreaming 1 \\\n",
"--test_tflite_nonstreaming_quantized 1 \\ # Test TFLite quantized non-streaming model\n", "--test_tflite_nonstreaming_quantized 1 \\\n",
"--test_tflite_streaming 1 \\ # Test TFLite streaming model\n", "--test_tflite_streaming 1 \\\n",
"--test_tflite_streaming_quantized 1 \\ # Test TFLite quantized streaming model\n", "--test_tflite_streaming_quantized 1 \\\n",
"--use_weights \"best_weights\" \\ # Use the best model weights for testing\n", "--use_weights \"best_weights\" \\\n",
"mixednet \\ # Specify the model architecture\n", "mixednet \\\n",
"--pointwise_filters \"64,96,128,160\" \\ # Optimized filter sizes\n", "--pointwise_filters \"64,96,128,160\" \\\n",
"--repeat_in_block \"2,2,3,3\" \\ # Increased repetitions for deeper feature learning\n", "--repeat_in_block \"2,2,3,3\" \\\n",
"--mixconv_kernel_sizes '[5], [7,11], [9,15], [17,23]' \\ # Wider kernels for better temporal feature extraction\n", "--mixconv_kernel_sizes '[5], [7,11], [9,15], [17,23]' \\\n",
"--residual_connection \"1,1,1,0\" \\ # Enable residuals for all but the last block\n", "--residual_connection \"1,1,1,0\" \\\n",
"--first_conv_filters 48 \\ # Larger initial filter size for improved feature extraction\n", "--first_conv_filters 48 \\\n",
"--first_conv_kernel_size 7 \\ # Larger kernel for the first convolution\n", "--first_conv_kernel_size 7 \\\n",
"--stride 2 # Reduce stride to preserve more temporal details\n" "--stride 2 \n"
] ]
}, },
{ {