linter and test updates for speaker_encoder, gmm_Attention

2026-09-10 11:50:20 +08:00 · 2019-11-12 12:42:42 +01:00
parent 1401a0db6b
commit df1b8b3ec7
14 changed files with 171 additions and 207 deletions
@@ -117,7 +117,8 @@ def format_data(data):

 def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler,
          ap, global_step, epoch):
-    data_loader = setup_loader(ap, model.decoder.r, is_val=False, verbose=(epoch == 0))
+    data_loader = setup_loader(ap, model.decoder.r, is_val=False,
+                               verbose=(epoch == 0))
    model.train()
    epoch_time = 0
    train_values = {