| 228 | ) |
| 229 | |
| 230 | def to_dict(self) -> Dict[str, Any]: |
| 231 | return { |
| 232 | "enabled": self.enabled, |
| 233 | "quantizer": { |
| 234 | name: cfg.to_dict() for name, cfg in self.quantizer.attributes.items() |
| 235 | }, |
| 236 | "entropy": { |
| 237 | "enabled": self.entropy.enabled, |
| 238 | "model_type": self.entropy.model_type, |
| 239 | "steps": dict(self.entropy.steps), |
| 240 | "factorized_lr": self.entropy.factorized_lr, |
| 241 | "gaussian_lr": self.entropy.gaussian_lr, |
| 242 | "scheduler_gamma": self.entropy.scheduler_gamma, |
| 243 | "rd_lambda": self.entropy.rd_lambda, |
| 244 | }, |
| 245 | "mask": self.mask.to_dict(), |
| 246 | } |