def METHOD_NAME(self, path, layer):
    """
    Saves centroids and assignments.
    Args:
        - path: folder used to save centroids and assignments
    """
    torch.METHOD_NAME(self.centroids, os.path.join(path, "{}_centroids.pth".format(layer)))
    torch.METHOD_NAME(
        self.assignments, os.path.join(path, "{}_assignments.pth".format(layer))
    )
    torch.METHOD_NAME(self.objective, os.path.join(path, "{}_objective.pth".format(layer)))