Skip to content

LLama3 dataset indexing fails #795

Description

@gyulaz-htec

I've downloaded the Llama3 dataset with the command from the readme: https://github.com/mlcommons/training/tree/master/large_language_model_pretraining/nemo#dataset.
I'm seeing the following error during build_data_index:

data_index/0 [default0]:[rank0]:   File "/opt/megatron-lm/megatron/core/datasets/blended_megatron_dataset_builder.py", line 126, in build                                                                 
data_index/0 [default0]:[rank0]:     datasets = self._build_blended_dataset_splits()                                                                                                                      
data_index/0 [default0]:[rank0]:   File "/opt/megatron-lm/megatron/core/datasets/blended_megatron_dataset_builder.py", line 259, in _build_blended_dataset_splits                                         
data_index/0 [default0]:[rank0]:     blended_datasets[i] = self._build_megatron_dataset_splits(                                                                                                           
data_index/0 [default0]:[rank0]:   File "/opt/megatron-lm/megatron/core/datasets/blended_megatron_dataset_builder.py", line 417, in _build_megatron_dataset_splits                                        
data_index/0 [default0]:[rank0]:     low_level_dataset = self.cls.build_low_level_dataset(dataset_path, self.config)                                                                                      
data_index/0 [default0]:[rank0]:   File "/opt/megatron-lm/megatron/core/datasets/gpt_dataset.py", line 150, in build_low_level_dataset                                                                    
data_index/0 [default0]:[rank0]:     return IndexedDataset(dataset_path, multimodal=False, mmap=config.mmap_bin_files)                                                                                    
data_index/0 [default0]:[rank0]:   File "/opt/megatron-lm/megatron/core/datasets/indexed_dataset.py", line 544, in __init__                                                                               
data_index/0 [default0]:[rank0]:     self.initialize(path_prefix, multimodal, mmap, s3_config)                                                                                                            
data_index/0 [default0]:[rank0]:   File "/opt/megatron-lm/megatron/core/datasets/indexed_dataset.py", line 584, in initialize                                                                             
data_index/0 [default0]:[rank0]:     self.index = _IndexReader(idx_path, self.multimodal)
data_index/0 [default0]:[rank0]:   File "/opt/megatron-lm/megatron/core/datasets/indexed_dataset.py", line 237, in __init__
data_index/0 [default0]:[rank0]:     with open(idx_path, "rb") as stream:
data_index/0 [default0]:[rank0]: IsADirectoryError: [Errno 21] Is a directory: '/data/data/c4-validation-91205-samples.en_text_document.idx'   

Most probably there is an issue with the folder name like it containing .-s. Could that confuses the file reading?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions