Skip to content

Commit

Permalink
fix xquant string adding if max cut was not computed
Browse files Browse the repository at this point in the history
  • Loading branch information
reuvenp committed Aug 13, 2024
1 parent 0ff61e7 commit c129bfa
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions model_compression_toolkit/xquant/common/tensorboard_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -115,12 +115,14 @@ def add_text_information(self,
similarity_metrics (Dict[str, Dict[str, float]]): A dictionary containing similarity metrics between quantized and float models for both representative and validation datasets.
quantized_model_metadata (Dict): Metadata from the quantized model.
"""
# Add the computed max cut
maxcut_str = f"MaxCut: {quantized_model_metadata['scheduling_info'][MAX_CUT]}"
self.tb_writer.add_text(maxcut_str, MAX_CUT)

# Add output similarity between quantized and float models on representative and validation datasets
output_similarity_repr = f"Similarity Metrics on outputs using representative dataset: \n" + "\n".join([f"{key}: {value:.4f}" for key, value in similarity_metrics[OUTPUT_SIMILARITY_METRICS_REPR].items()])
output_similarity_val = f"Similarity Metrics on outputs using validation dataset: \n" + "\n".join([f"{key}: {value:.4f}" for key, value in similarity_metrics[OUTPUT_SIMILARITY_METRICS_VAL].items()])
self.tb_writer.add_text(output_similarity_repr, OUTPUT_SIMILARITY_METRICS_REPR)
self.tb_writer.add_text(output_similarity_val, OUTPUT_SIMILARITY_METRICS_VAL)

# Add the max cut if it was computed
if 'scheduling_info' in quantized_model_metadata:
maxcut_str = f"MaxCut: {quantized_model_metadata['scheduling_info'][MAX_CUT]}"
self.tb_writer.add_text(maxcut_str, MAX_CUT)

0 comments on commit c129bfa

Please sign in to comment.