HoneyTian commited on
Commit
6e41e8e
·
1 Parent(s): ccf5554
examples/conv_tasnet/step_2_train_model.py CHANGED
@@ -240,7 +240,6 @@ def main():
240
  total_lds_loss = 0.
241
  total_batches = 0.
242
  progress_bar = tqdm(
243
- total=len(train_data_loader),
244
  desc="Training; epoch: {}".format(idx_epoch),
245
  )
246
  for batch in train_data_loader:
@@ -306,7 +305,6 @@ def main():
306
  total_batches = 0.
307
 
308
  progress_bar = tqdm(
309
- total=len(valid_data_loader),
310
  desc="Evaluation; epoch: {}".format(idx_epoch),
311
  )
312
  with torch.no_grad():
 
240
  total_lds_loss = 0.
241
  total_batches = 0.
242
  progress_bar = tqdm(
 
243
  desc="Training; epoch: {}".format(idx_epoch),
244
  )
245
  for batch in train_data_loader:
 
305
  total_batches = 0.
306
 
307
  progress_bar = tqdm(
 
308
  desc="Evaluation; epoch: {}".format(idx_epoch),
309
  )
310
  with torch.no_grad():