Skip to content
Permalink

Comparing changes

This is a direct comparison between two commits made in this repository or its related repositories. View the default comparison for this range or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: sony/model_optimization
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: 4e901775b653d1adbbd00d549461931d94a35b07
Choose a base ref
..
head repository: sony/model_optimization
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: 949b7868695434f3ba57fc84a38c5cdd810f1478
Choose a head ref
Showing with 1 addition and 8 deletions.
  1. +1 −1 model_compression_toolkit/core/runner.py
  2. +0 −7 tests/keras_tests/feature_networks_tests/feature_networks/mixed_precision_tests.py
2 changes: 1 addition & 1 deletion model_compression_toolkit/core/runner.py
Original file line number Diff line number Diff line change
@@ -223,7 +223,7 @@ def _set_final_resource_utilization(graph: Graph,
fw_impl: FrameworkImplementation object with specific framework methods implementation.
"""
ru_targets = target_resource_utilization.get_restricted_targets()
ru_targets = target_resource_utilization.get_restricted_targets() if target_resource_utilization else None
final_ru = None
if ru_targets:
ru_calculator = ResourceUtilizationCalculator(graph, fw_impl, fw_info)
Original file line number Diff line number Diff line change
@@ -484,13 +484,6 @@ def compare(self, quantized_model, float_model, input_x=None, quantization_info:
activation_layers_idx=self.activation_layers_idx,
unique_tensor_values=16)

# Verify final ResourceUtilization
self.unit_test.assertTrue(
quantization_info.final_resource_utilization.total_memory ==
quantization_info.final_resource_utilization.weights_memory + quantization_info.final_resource_utilization.activation_memory,
"Running weights and activation mixed-precision, "
"final total memory should be equal to sum of weights and activation memory.")


class MixedPrecisionMultipleResourcesTightUtilizationSearchTest(MixedPrecisionActivationBaseTest):
def __init__(self, unit_test):