Method default_config

axlearn/common/quantizer.py:116–122

(cls)

    @classmethod
    def default_config(cls):
        cfg: BaseQuantizer.Config = super().default_config()
        # Do not shard the codebook.
        # ToDo(zhiyunlu): investigate the codebook partition_spec.
        cfg.param_partition_spec = (None, None, None)
        cfg.param_init = REQUIRED
        return cfg

    def _create_layer_parameter_specs(self) -> dict[str, ParameterSpec]:
        cfg = self.config