🐞 Fix EfficientAD's pretrained weigths load path (#1966)

seyeon923 · web-flow · commit 25fc952830aa · 2024-04-09T09:14:28.000+01:00
Fix EfficientAD's pretrained weigths load path to be identical to downloaded path
diff --git a/src/anomalib/models/image/efficient_ad/lightning_model.py b/src/anomalib/models/image/efficient_ad/lightning_model.py
@@ -93,7 +93,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)))

Original file line number	Diff line number	Diff line change
`@@ -93,7 +93,7 @@ def prepare_pretrained_model(self) -> None:`
`93`	`93`	`if not (pretrained_models_dir / "efficientad_pretrained_weights").is_dir():`
`94`	`94`	`download_and_extract(pretrained_models_dir, WEIGHTS_DOWNLOAD_INFO)`
`95`	`95`	`teacher_path = (`
`96`		`- pretrained_models_dir / "efficientad_pretrained_weights" / f"pretrained_teacher_{self.model_size}.pth"`
	`96`	`+ pretrained_models_dir / "efficientad_pretrained_weights" / f"pretrained_teacher_{self.model_size.value}.pth"`
`97`	`97`	`)`
`98`	`98`	`logger.info(f"Load pretrained teacher model from {teacher_path}")`
`99`	`99`	`self.model.teacher.load_state_dict(torch.load(teacher_path, map_location=torch.device(self.device)))`