resolve workdir for load_model during testing

lupus83 · Mar 30, 2023 · 8c5bae9 · 8c5bae9
1 parent e7934f4
commit 8c5bae9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchpanic/modules/utils.py b/torchpanic/modules/utils.py
@@ -79,7 +79,7 @@ def load_config(ckpt_path: Union[str, Path]) -> DictConfig:
     with open(config_path) as f:
         y = yaml_load(f)
     workdir = Path().absolute()
-    idx = workdir.parts.index('panic')
+    idx = workdir.parts.index('PANIC')
     workdir = Path(*workdir.parts[:idx+1])
     if 'protonet' in y['model']['net']:
         y['model']['net']['protonet']['pretrained_model'] = \