        pbar = tqdm.tqdm(
            enumerate(train_loader),
            total=len(train_loader),
            colour="blue",
            desc=f"Epoch {curr_epoch}",
            disable=(local_rank != 0),
        )