[Python] Progress bar 사용하기

1. 라이브러리 선언

• argparse 사용을 위해 라이브러리 선언하기

from tqdm.notebook import tqdm

1. Argparse 코드

pbar = tqdm(range(opt.epochs), unit = 'epoch')
        for epoch in pbar:
            # 본인이 사용하는 코드
            
            pbar.set_postfix({'y_loss' : y_loss, 'd_acc' : d_acc}

• for문을 다음과 같이 바꾸면 예쁜 progress bar가 탄생합니다.

• 저는 Domain adaptation 연구를 하고 있어 현재 라벨의 손실값과 도메인의 정확도를 postfix로 놔두었습니다.

  postfix에 대해 자세한 것은 코드와 아래 사진을 비교하면서 이해하시면 되겠습니다.

• unit에는 자신이 사용하는 단위를 적어주면 됩니다.

• 이에 대한 결과는 다음과 같습니다.

• 포스트는 Window10 기준으로 작성되었습니다

DL Code 카테고리 내 다른 글 보러가기