def compare_and_validate_results(float_model, q_model):
     weight_dict = compare_weights_fx(float_model.state_dict(),
                                      q_model.state_dict())
     self.assertEqual(len(weight_dict), 1)
     for k, v in weight_dict.items():
         self.assertTrue(len(v["float"]) == len(v["quantized"]))
         for i, val in enumerate(v["quantized"]):
             self.assertTrue(
                 v["float"][i].shape == v["quantized"][i].shape)
Esempio n. 2
0
    def compare_and_validate_model_weights_results_fx(
        self, float_model, q_model, expected_weight_dict_keys
    ):
        weight_dict = compare_weights_fx(float_model.state_dict(), q_model.state_dict())

        self.assertTrue(weight_dict.keys() == expected_weight_dict_keys)
        self.assertEqual(len(weight_dict), 1)

        for k, v in weight_dict.items():
            self.assertTrue(v["float"].shape == v["quantized"].shape)