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)))