fixed two params (#108)

VMarsocci · Oct 23, 2024 · 102c575 · 102c575
1 parent 4d3c736
commit 102c575
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pangaea/datasets/fivebillionpixels.py b/pangaea/datasets/fivebillionpixels.py
@@ -102,7 +102,7 @@ def __init__(
         self.auto_download = auto_download
 
         self._image_dir = sorted(
-            glob(os.path.join(self._base_dir, self.split, "imgs", "*.tif"))
+            glob(os.path.join(self._base_dir, self.split, "images", "*.tif"))
         )
         self._label_dir = sorted(
             glob(os.path.join(self._base_dir, self.split, "labels", "*.tif"))

diff --git a/pangaea/run.py b/pangaea/run.py
@@ -249,7 +249,7 @@ def main(cfg: DictConfig) -> None:
         )
         # resume training if model_checkpoint is provided
         if cfg.ckpt_dir is not None:
-            trainer.load_model(cfg.resume_from)
+            trainer.load_model(cfg.ckpt_dir)
 
         trainer.train()