Method clear

ppq/samples/QAT/trainer.py:142–146

Clear training state.

(self)

        exporter.export(file_path=file_path, graph=self.graph)

    def clear(self):
        """Clear training state."""
        for tensor in self._training_graph.parameters():
            tensor.requires_grad = False
            tensor._grad = None