fix params in decoder (#162)

2025-12-19 09:44:19 +01:00 · 2022-06-22 16:45:01 -05:00
parent c8422ffd5d
commit fddf66e91e
1 changed files with 2 additions and 2 deletions
--- a/train_decoder.py
+++ b/train_decoder.py
@@ -258,8 +258,8 @@ def train(
    is_master = accelerator.process_index == 0

    trainer = DecoderTrainer(
-        accelerator,
-        decoder,
+        decoder=decoder,
+        accelerator=accelerator,
        **kwargs
    )