🐞 Fix category check for folder dataset in anomalib CLI (#567)

* Fix category check * Fix config file
openvinotoolkit · Sep 20, 2022 · 4860abc · 4860abc
1 parent a055416
commit 4860abc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/anomalib/utils/cli/cli.py b/anomalib/utils/cli/cli.py
@@ -144,7 +144,7 @@ def __set_default_root_dir(self) -> None:
             root_dir = config.trainer.default_root_dir if config.trainer.default_root_dir else "./results"
             model_name = config.model.class_path.split(".")[-1].lower()
             data_name = config.data.class_path.split(".")[-1].lower()
-            category = config.data.init_args.category if config.data.init_args.keys() else ""
+            category = config.data.init_args.category if "category" in config.data.init_args else ""
             time_stamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
             default_root_dir = os.path.join(root_dir, model_name, data_name, category, time_stamp)
 

diff --git a/configs/model/fastflow.yaml b/configs/model/fastflow.yaml
@@ -30,7 +30,7 @@ model:
     hidden_ratio: 1.0 # options: [1.0, 1.0, 0.16, 0.16] - for each supported backbone
 
 optimizer:
-  class_path: torch.optim._multi_tensor.Adam
+  class_path: torch.optim._multi_tensor.adam.Adam
   init_args:
     lr: 0.001
     weight_decay: 0.00001