facebookresearch/fairseq
1
1
Fork 0
mirror of https://github.com/facebookresearch/fairseq.git synced 2024-09-11 17:25:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

wav2vec everstore support fix

Browse Source 
Summary: fixes some merge issues that prevented wav2vec from training properly

Reviewed By: myleott

Differential Revision: D16981120

fbshipit-source-id: cad39aaf2f44daabcbafe7b4e8735d055b3842a7
This commit is contained in:
Alexei Baevski 2019-08-26 18:30:17 -07:00 committed by Facebook Github Bot
parent 8a8c0691ba
commit 3ab8e0fd73
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -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