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