check pythonpath

2022-09-28 11:51:58 +05:30 · 2022-09-28 11:51:58 +05:30 · 38e38dc7df
parent 3570beb2d8
commit 38e38dc7df
1 changed files with 1 additions and 0 deletions
--- a/cli/train.py
+++ b/cli/train.py
@ -4,6 +4,7 @@ from omegaconf import DictConfig
 from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
 from pytorch_lightning.loggers import MLFlowLogger

+from enhancer.data.dataset import EnhancerDataset

@hydra.main(config_path="train_config",config_name="config")
 def main(config: DictConfig):