check pythonpath
This commit is contained in:
parent
3570beb2d8
commit
38e38dc7df
|
|
@ -4,6 +4,7 @@ from omegaconf import DictConfig
|
||||||
from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
|
from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
|
||||||
from pytorch_lightning.loggers import MLFlowLogger
|
from pytorch_lightning.loggers import MLFlowLogger
|
||||||
|
|
||||||
|
from enhancer.data.dataset import EnhancerDataset
|
||||||
|
|
||||||
@hydra.main(config_path="train_config",config_name="config")
|
@hydra.main(config_path="train_config",config_name="config")
|
||||||
def main(config: DictConfig):
|
def main(config: DictConfig):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue