Skip to content

Remove local_rank and idist.barrier() from data.py if no distributed configuration selected #168

@vfdev-5

Description

@vfdev-5

Clear and concise description of the problem

Got a feedback that this code

    local_rank = idist.get_local_rank()

    ...

    if local_rank > 0:
        # Ensure that only rank 0 download the dataset
        idist.barrier()

    ...
    
    )
    if local_rank == 0:
        # Ensure that only rank 0 download the dataset
        idist.barrier()

looks a bit strange if no distributed configuration is selected

Let's put template conditions here as well

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions