Convenience static method wrapping calibrate.py.
(
model: nn.Module,
tokenizer,
dataset: str = "wikitext",
num_samples: int = 2000,
save_path: str = "./router.pt",
config: Optional[TIDEConfig] = None,
)
| 373 | |
| 374 | @staticmethod |
| 375 | def calibrate( |
| 376 | model: nn.Module, |
| 377 | tokenizer, |
| 378 | dataset: str = "wikitext", |
| 379 | num_samples: int = 2000, |
| 380 | save_path: str = "./router.pt", |
| 381 | config: Optional[TIDEConfig] = None, |
| 382 | ) -> RouterCheckpoint: |
| 383 | """Convenience static method wrapping calibrate.py.""" |
| 384 | from TIDE.calibrate import calibrate as _calibrate |
| 385 | |
| 386 | cfg = config or TIDEConfig() |
| 387 | cfg.calibration_samples = num_samples |
| 388 | cfg.calibration_dataset = dataset |
| 389 | return _calibrate(model, tokenizer, config=cfg, save_path=save_path) |