def load(self, path, layer):
        """
        Loads centroids and assignments from a given path

        Args:
            - path: folder use to load centroids and assignments
        """

        self.centroids = torch.load(
            os.path.join(path, "{}_centroids.pth".format(layer))
        )
        self.assignments = torch.load(
            os.path.join(path, "{}_assignments.pth".format(layer))
        )
        self.objective = torch.load(
            os.path.join(path, "{}_objective.pth".format(layer))
        )

        Loads centroids and assignments from a given path

        Args:
            - path: folder use to load centroids and assignments
        
structured_prompting/fairseq-version/fairseq/fairseq/modules/quantization/pq/em.py