Fix import paths

Clay-foundation · May 24, 2024 · b83305b · b83305b
1 parent a091e51
commit b83305b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/train_clay.sh b/train_clay.sh
@@ -20,4 +20,4 @@ export PYTHONFAULTHANDLER=1
 export PYTHONUNBUFFERED=1
 
 # Run the model on multi node parallel cluster
-srun python trainer.py fit --model ClayMAEModule --data ClayDataModule --config configs/config.yaml --data.data_dir /fsx/dra1/naip
+srun python trainer.py fit --model ClayMAEModule --data ClayDataModule --config configs/config.yaml --data.data_dir /fsx
diff --git a/trainer.py b/trainer.py
@@ -13,7 +13,7 @@
 from lightning.pytorch.cli import LightningCLI
 
 from src.datamodule import ClayDataModule  # noqa: F401
-from src.model_clay_v1 import ClayMAEModule  # noqa: F401
+from src.model import ClayMAEModule  # noqa: F401
 
 
 # %%