ModuleNotFoundError: No module named 'dataset'

Traceback (most recent call last):
File "/home/zlj/Ask-Anything/video_chat2/tasks/train_qformer.py", line 13, in
from dataset import create_dataset, create_loader, create_sampler, Metaloader
ModuleNotFoundError: No module named 'dataset'
Traceback (most recent call last):
File "/home/zlj/Ask-Anything/video_chat2/tasks/train_qformer.py", line 13, in
from dataset import create_dataset, create_loader, create_sampler, Metaloader
ModuleNotFoundError: No module named 'dataset'
Traceback (most recent call last):
File "/home/zlj/Ask-Anything/video_chat2/tasks/train_qformer.py", line 13, in
from dataset import create_dataset, create_loader, create_sampler, Metaloader
ModuleNotFoundError: No module named 'dataset'
Traceback (most recent call last):
File "/home/zlj/Ask-Anything/video_chat2/tasks/train_qformer.py", line 13, in
from dataset import create_dataset, create_loader, create_sampler, Metaloader
ModuleNotFoundError: No module named 'dataset'
Traceback (most recent call last):
File "/home/zlj/Ask-Anything/video_chat2/tasks/train_qformer.py", line 13, in
from dataset import create_dataset, create_loader, create_sampler, Metaloader
ModuleNotFoundError: No module named 'dataset'
Traceback (most recent call last):
File "/home/zlj/Ask-Anything/video_chat2/tasks/train_qformer.py", line 13, in
from dataset import create_dataset, create_loader, create_sampler, Metaloader
ModuleNotFoundError: No module named 'dataset'
Traceback (most recent call last):
File "/home/zlj/Ask-Anything/video_chat2/tasks/train_qformer.py", line 13, in
from dataset import create_dataset, create_loader, create_sampler, Metaloader
ModuleNotFoundError: No module named 'dataset'
Traceback (most recent call last):
File "/home/zlj/Ask-Anything/video_chat2/tasks/train_qformer.py", line 13, in
from dataset import create_dataset, create_loader, create_sampler, Metaloader
ModuleNotFoundError: No module named 'dataset'
ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 1) local_rank: 0 (pid: 1250378) of binary: /home/zlj/miniconda3/envs/videochat2/bin/python
Traceback (most recent call last):
File "/home/zlj/miniconda3/envs/videochat2/bin/torchrun", line 8, in
sys.exit(main())
File "/home/zlj/miniconda3/envs/videochat2/lib/python3.10/site-packages/torch/distributed/elastic/multiprocessing/errors/init.py", line 346, in wrapper
return f(*args, **kwargs)
File "/home/zlj/miniconda3/envs/videochat2/lib/python3.10/site-packages/torch/distributed/run.py", line 762, in main
run(args)
File "/home/zlj/miniconda3/envs/videochat2/lib/python3.10/site-packages/torch/distributed/run.py", line 753, in run
elastic_launch(
File "/home/zlj/miniconda3/envs/videochat2/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 132, in call
return launch_agent(self._config, self._entrypoint, list(args))
File "/home/zlj/miniconda3/envs/videochat2/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 246, in launch_agent
raise ChildFailedError(
torch.distributed.elastic.multiprocessing.errors.ChildFailedError:

tasks/train_qformer.py FAILED

Start with tasks/train_qformer.py, especially its dataset import and the command used to launch it with torchrun. Trace how the dataset module is expected to be discovered, then verify that the training entry point gets past the import under the documented environment. Done means the reported ModuleNotFoundError no longer occurs when launching this task.