Added progress bar support

DevKiHyun · Aug 24, 2017 · 683eabe · 683eabe
1 parent 7025ec6
commit 683eabe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -3,3 +3,4 @@ torch
 visdom
 wget
 librosa
+tqdm
diff --git a/test.py b/test.py
@@ -1,6 +1,8 @@
 import argparse
 
 from torch.autograd import Variable
+from tqdm import tqdm
+
 from decoder import GreedyDecoder, BeamCTCDecoder
 
 from data.data_loader import SpectrogramDataset, AudioDataLoader
@@ -46,7 +48,7 @@
  test_loader = AudioDataLoader(test_dataset, batch_size=args.batch_size,
  num_workers=args.num_workers)
  total_cer, total_wer = 0, 0
- for i, (data) in enumerate(test_loader):
+ for i, (data) in tqdm(enumerate(test_loader), total=len(test_loader)):
  inputs, targets, input_percentages, target_sizes = data
 
  inputs = Variable(inputs, volatile=True)