Fixed horovod backend

ludwig-ai · Dec 31, 2022 · b99e580 · b99e580
1 parent bee031b
commit b99e580
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ludwig/backend/horovod.py b/ludwig/backend/horovod.py
@@ -44,7 +44,7 @@ def initialize(self):
 
     def initialize_pytorch(self, *args, **kwargs):
         initialize_pytorch(
-            *args, local_rank=self._distributed.local_rank(), local_size=self._distributed.local_size() ** kwargs
+            *args, local_rank=self._distributed.local_rank(), local_size=self._distributed.local_size(), **kwargs
         )
 
     def create_trainer(self, **kwargs) -> "BaseTrainer":  # noqa: F821