diff --git a/src/ptbench/engine/trainer.py b/src/ptbench/engine/trainer.py index 14e815b0390ebab929799e0de263c69a22ae6fb9..b8532e85a0e488a703225a70ddbeab0db42d689a 100644 --- a/src/ptbench/engine/trainer.py +++ b/src/ptbench/engine/trainer.py @@ -189,7 +189,7 @@ def run( save a checkpoint every ``n`` epochs. If set to ``0`` (zero), then do not save intermediary checkpoints - accelerator : str` + accelerator : str accelerator to use arguments : dict