Ejemplos de Config.set_quantization_dataloader en Python

Ejemplo n.º 1

Archivo: test_config.py Proyecto: vuiseng9/lpot

    def test_set_quantization_dataloader_on_empty_config(self) -> None:
        """Test set_quantization_dataloader on empty config."""
        config = Config()

        config.set_quantization_dataloader({
            "name": "dataloader_name",
        }, )

        self.assertIsNone(config.quantization)

Ejemplo n.º 2

Archivo: test_config.py Proyecto: vuiseng9/lpot

    def test_set_quantization_dataloader(self) -> None:
        """Test set_quantization_dataloader."""
        config = Config(self.predefined_config)

        config.set_quantization_dataloader({
            "name": "dataloader_name",
        }, )

        self.assertEqual(
            "dataloader_name",
            config.quantization.calibration.dataloader.dataset.name,
        )