Fix broken progress bar (#524)
Fix broken progress bar (#524)

- Fix broken update calculation
- Make progress bar use the neat `unit_scale` feature
  of tqdm
pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision