Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How can I train on nuscenes v1.0-mini? #19

Open
zheng0819 opened this issue Mar 24, 2023 · 1 comment
Open

How can I train on nuscenes v1.0-mini? #19

zheng0819 opened this issue Mar 24, 2023 · 1 comment

Comments

@zheng0819
Copy link

The memory of my computer is not enough to train on nuscenes v1.0, so I try to train on nuscenes v1.0-mini. I change the config file, but an error occurred while training :
Caught ValueError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/yusen0819/anaconda3/envs/vista/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 202, in _worker_loop
data = fetcher.fetch(index)
File "/home/yusen0819/anaconda3/envs/vista/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/yusen0819/anaconda3/envs/vista/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/yusen0819/VISTA/det3d/datasets/nuscenes/nuscenes.py", line 193, in getitem
return self.get_sensor_data(idx)
File "/home/yusen0819/VISTA/det3d/datasets/nuscenes/nuscenes.py", line 189, in get_sensor_data
data, _ = self.pipeline(res, info)
File "/home/yusen0819/VISTA/det3d/datasets/pipelines/compose.py", line 25, in call
res, info = t(res, info)
File "/home/yusen0819/VISTA/det3d/datasets/pipelines/preprocess.py", line 96, in call
gt_drop_max_keep=self.gt_drop_max_keep,
File "/home/yusen0819/VISTA/det3d/core/sampler/sample_ops.py", line 247, in sample_all
"points": np.concatenate(s_points_list, axis=0),
File "<array_function internals>", line 6, in concatenate
ValueError: need at least one array to concatenate
File "/home/yusen0819/VISTA/det3d/torchie/trainer/trainer.py", line 389, in train
for i, data_batch in enumerate(data_loader):
File "/home/yusen0819/VISTA/det3d/torchie/trainer/trainer.py", line 542, in run
epoch_runner(data_loaders[i], self.epoch, **kwargs)
File "/home/yusen0819/VISTA/det3d/torchie/apis/train.py", line 336, in train_detector
trainer.run(data_loaders, cfg.workflow, cfg.total_epochs, local_rank=cfg.local_rank)
File "/home/yusen0819/VISTA/tools/train.py", line 126, in main
logger=logger,
File "/home/yusen0819/VISTA/tools/train.py", line 131, in
main()
I cannot handle this error. How can I train on nuscenes v1.0-mini? Thanks.

@AndlollipopDE
Copy link
Member

We have not tried to train on the mini version, but the error messages indicate that you may not generate data properly (the database sampling).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants