ModuleNotFoundError: No module named 'train'

[HaoranDeng]

Hi,

When running the training script with Hydra configuration, an import error occurs indicating that the module train cannot be found.

(venv) (base) denghaoran@scai3:~/workspace/tmp/based/train$ python run.py experiment=example/based-360m trainer.devices=1
/home/denghaoran/workspace/tmp/based/train/run.py:33: UserWarning:
The version_base parameter is not specified.
Please specify a compatability version level, or None.
Will assume defaults for version 1.1
  @hydra.main(config_path="configs/", config_name="config.yaml")
/home/denghaoran/workspace/based/venv/lib/python3.10/site-packages/hydra/_internal/hydra.py:119: UserWarning: Future Hydra versions will no longer change working directory at job runtime by default.
See https://hydra.cc/docs/1.2/upgrades/1.1_to_1.2/changes_to_job_working_dir/ for more information.
  ret = run_job(
Error executing job with overrides: ['experiment=example/based-360m', 'trainer.devices=1']
Traceback (most recent call last):
  File "/home/denghaoran/workspace/tmp/based/train/run.py", line 42, in main
    from train.training import train
ModuleNotFoundError: No module named 'train'

Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.

Thanks in advance:)