You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

evaluate_interpretations.py 590B

123456789101112131415161718192021
  1. from torchlap.experiments._entry_loader import get_entry, PhaseType
  2. from torchlap.experiments._model_loading import load_model
  3. from torchlap.interpreting.interpretation_evaluation_runner import InterpretingEvalRunner
  4. from torchlap.utils.hooker import Hooker
  5. def main() -> None:
  6. ep = get_entry(PhaseType.EVALINTERPRET)
  7. conf = ep.conf
  8. model = ep.model
  9. model = load_model(model, conf)
  10. model.eval()
  11. runner = InterpretingEvalRunner(conf, model)
  12. with Hooker(*conf.hooks_by_phase[conf.phase_type]):
  13. runner.evaluate()
  14. if __name__ == '__main__':
  15. main()