Update trainer.py

lifeiteng · Apr 19, 2023 · 27c0667 · 27c0667
1 parent 1cd9b8a
commit 27c0667
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/valle/bin/trainer.py b/valle/bin/trainer.py
@@ -635,7 +635,7 @@ def train_one_epoch(
     elif params.dtype in ["float16", "fp16"]:
         dtype, enabled = torch.float16, True
 
-    model_context = model.join isinstance(model, DDP) else nullcontext
+    model_context = model.join if isinstance(model, DDP) else nullcontext
     with model_context():
         batch_idx = 0
         while True: