Skip to content

Commit 8d80c33

Browse files
Fix parameter name for TPC version to SDSP version in quantization configurations
1 parent ece68e3 commit 8d80c33

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tutorials/notebooks/mct_features_notebooks/keras/example_keras_mct_wrapper.ipynb

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@
317317
"\n",
318318
" # Parameter configuration for PTQ\n",
319319
" param_items = [\n",
320-
" ['target_platform_version', 'v1'], # The version of the TPC to use.\n",
320+
" ['sdsp_version', '3.14'], # The version of the SDSP converter.\n",
321321
" ['activation_error_method', QuantizationErrorMethod.MSE], # Error metric for activation.\n",
322322
" ['weights_bias_correction', True], # Enable bias correction for weights.\n",
323323
" ['z_threshold', float('inf')], # Threshold for zero-point quantization.\n",
@@ -382,7 +382,7 @@
382382
"\n",
383383
" # Parameter configuration for PTQ with Mixed Precision\n",
384384
" param_items = [\n",
385-
" ['target_platform_version', 'v1'], # The version of the TPC to use.\n",
385+
" ['sdsp_version', '3.14'], # The version of the SDSP converter.\n",
386386
" ['num_of_images', 5], # Number of epochs for gradient-based fine-tuning.\n",
387387
" ['use_hessian_based_scores', False], # Use Hessian-based sensitivity scores for layer importance.\n",
388388
" ['weights_compression_ratio', 0.75], # Target compression ratio for model weights (75% of original size.\n",
@@ -446,7 +446,7 @@
446446
" # Parameter configuration for GPTQ\n",
447447
" param_items = [\n",
448448
" # Platform configuration\n",
449-
" ['target_platform_version', 'v1'], # The version of the TPC to use.\n",
449+
" ['sdsp_version', '3.14'], # The version of the SDSP converter.\n",
450450
" ['n_epochs', 5], # Number of epochs for gradient-based fine-tuning.\n",
451451
" ['optimizer', None], # Optimizer to use during fine-tuning.\n",
452452
" ['save_model_path', './qmodel_GPTQ_Keras.keras'] # Path to save the quantized model.\n",
@@ -512,7 +512,7 @@
512512
" # Parameter configuration for GPTQ with Mixed Precision\n",
513513
" param_items = [\n",
514514
" # Platform configuration\n",
515-
" ['target_platform_version', 'v1'], # The version of the TPC to use.\n",
515+
" ['sdsp_version', '3.14'], # The version of the SDSP converter.\n",
516516
" ['n_epochs', 5], # Number of epochs for gradient-based fine-tuning.\n",
517517
" ['optimizer', None], # Optimizer to use during fine-tuning.\n",
518518
" ['num_of_images', 5], # Number of images to use for calibration.\n",

0 commit comments

Comments
 (0)