From 6f7e210e4813e264bd9ab4141c82c731e74ad466 Mon Sep 17 00:00:00 2001 From: Kim Se-yeon Date: Tue, 9 Apr 2024 14:07:46 +0900 Subject: [PATCH] Fix EfficientAD's pretrained weigths load path to be identical to downloaded path --- src/anomalib/models/image/efficient_ad/lightning_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/anomalib/models/image/efficient_ad/lightning_model.py b/src/anomalib/models/image/efficient_ad/lightning_model.py index 1f4026a8ac..baffd0c2ad 100644 --- a/src/anomalib/models/image/efficient_ad/lightning_model.py +++ b/src/anomalib/models/image/efficient_ad/lightning_model.py @@ -125,7 +125,7 @@ def prepare_pretrained_model(self) -> None: if not (pretrained_models_dir / "efficientad_pretrained_weights").is_dir(): download_and_extract(pretrained_models_dir, WEIGHTS_DOWNLOAD_INFO) teacher_path = ( - pretrained_models_dir / "efficientad_pretrained_weights" / f"pretrained_teacher_{self.model_size}.pth" + pretrained_models_dir / "efficientad_pretrained_weights" / f"pretrained_teacher_{self.model_size.value}.pth" ) logger.info(f"Load pretrained teacher model from {teacher_path}") self.model.teacher.load_state_dict(torch.load(teacher_path, map_location=torch.device(self.device)))