make dir before save anomaly weight

openvinotoolkit · Oct 27, 2022 · 4892b19 · 4892b19
1 parent fd16958
commit 4892b19
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ote_cli/ote_cli/utils/hpo.py b/ote_cli/ote_cli/utils/hpo.py
@@ -517,6 +517,7 @@ def __init__(
             if _is_anomaly_framework_task(task_type):
                 impl_class = get_impl_class(environment.model_template.entrypoints.base)
                 task = impl_class(task_environment=environment)
+                os.makedirs(self.work_dir, exist_ok=True)
                 torch.save(task.model_info(), osp.join(self.work_dir, "weights.pth"))
         else:
             save_model_data(environment.model, self.work_dir)