(cfg)
| 8 | |
| 9 | @hydra.main(config_path="../config", config_name="base", version_base="1.2") |
| 10 | def main(cfg): |
| 11 | os.environ["CUDA_VISIBLE_DEVICES"] = cfg.retrieval.cuda_visible_devices |
| 12 | retrieval_class_path = cfg.retrieval.class_path |
| 13 | module_name, class_name = retrieval_class_path.rsplit('.', 1) |
| 14 | module = importlib.import_module(module_name) |
| 15 | retrieval_class = getattr(module, class_name) |
| 16 | |
| 17 | dataset = BaseDataset(cfg.dataset) |
| 18 | retrieval_model:BaseRetrieval = retrieval_class(cfg.retrieval) |
| 19 | retrieval_model.find_top_k(dataset) |
| 20 | |
| 21 | if __name__ == "__main__": |
| 22 | main() |
no test coverage detected