Skip to content

Commit

Permalink
Remove horovod option from RayNoiseGeneratorTrainer
Browse files Browse the repository at this point in the history
  • Loading branch information
annaelisalappe authored Nov 26, 2024
1 parent 8c27729 commit 765b33c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion use-cases/virgo/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -375,7 +375,7 @@ class RayNoiseGeneratorTrainer(RayTorchTrainer):
def __init__(
self,
config: Dict,
strategy: Optional[Literal["ddp", "deepspeed", "horovod"]] = "ddp",
strategy: Optional[Literal["ddp", "deepspeed"]] = "ddp",
name: Optional[str] = None,
logger: Optional[Logger] = None,
) -> None:
Expand Down

0 comments on commit 765b33c

Please sign in to comment.