How to train with part of the whole dataset?

[jarvishou829]

Thanks for the great job. I wonder how to train with part of a whole big dataset like nuscenes. Because the trainval is too large and mini is too small. I wonder if I can spilt a quarter of the trainval dataset for training.

[ZwwWayne]

Hi @jarvishou829 ,
You can try MMDet3D 1.x and simply setting indices=N in the dataset can allow you to train the model only by N samples of the dataset.

[jarvishou829]

Thanks for your fast reply. Is the indices an arg of mmdet.datasets.DATASETS? I initialize the NuScenesDataset and register it to the mmdet.datasets.DATASETS, so should I setindicesin the init of NuScenesDataset?

[JingweiZhang12]

indices is an argument of mmengine.BaseDataset. https://github.com/open-mmlab/mmengine/blob/main/mmengine/dataset/base_dataset.py#L219

[jarvishou829]

That helps a lot! I will check and try it. Thanks.