[PyTorch] 학습 progress bar 설정하기

학습시에 진행과정, 경과시간, 예상시간을 확인하고 싶은 경우, 사용가능한 함수가 mmcv에 구현되어 있다.

 

import mmcv

prog_bar = mmcv.ProgressBar(len(dataset))

for i, data in enumerate(data_loader):
	...
    
    for _ in range(batch_size):
    	prog_bar.update()