def METHOD_NAME(self, state_dict):
    self.n_centroids_config = state_dict["n_centroids_config"]
    self.block_sizes_config = state_dict["block_sizes_config"]
    self.layers_to_quantize = state_dict["layers_to_quantize"]
    self.epoch_schedule = state_dict["epoch_schedule"]
    self.update_schedule = state_dict["update_schedule"]
    self.quantization_step = state_dict["quantization_step"]