MCPcopy Create free account
hub / github.com/apple/axlearn / Config

Class Config

axlearn/common/quantizer.py:569–572  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

567
568 @config_class
569 class Config(BaseQuantizer.Config):
570 input_dim: Required[int] = REQUIRED
571 input_proj: MultiLinear.Config = MultiLinear.default_config()
572 temperature_schedule: Required[InstantiableConfig] = REQUIRED
573
574 def __init__(self, cfg: Config, *, parent: Optional[Module]):
575 super().__init__(cfg, parent=parent)

Callers

nothing calls this directly

Calls 1

default_configMethod · 0.45

Tested by

no test coverage detected