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