Skip to content

Commit

Permalink
wav2vec everstore support fix
Browse files Browse the repository at this point in the history
Summary: fixes some merge issues that prevented wav2vec from training properly

Reviewed By: myleott

Differential Revision: D16981120

fbshipit-source-id: cad39aaf2f44daabcbafe7b4e8735d055b3842a7
  • Loading branch information
Alexei Baevski authored and facebook-github-bot committed Aug 27, 2019
1 parent 8a8c069 commit 3ab8e0f
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions train.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,6 +117,10 @@ def train(args, trainer, task, epoch_itr):
valid_subsets = args.valid_subset.split(',')
max_update = args.max_update or math.inf
for i, samples in enumerate(progress, start=epoch_itr.iterations_in_epoch):
samples = [s for s in samples if len(s) > 0]
if len(samples) == 0:
continue

log_output = trainer.train_step(samples)
if log_output is None:
continue
Expand Down

0 comments on commit 3ab8e0f

Please sign in to comment.