Skip to content

Commit

Permalink
Fix docstring (#447)
Browse files Browse the repository at this point in the history
  • Loading branch information
muellerzr authored Jun 15, 2022
1 parent 873dcc6 commit fa17f20
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/accelerate/checkpointing.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,7 @@ def load_accelerator_state(input_dir, models, optimizers, schedulers, process_in
Args:
input_dir (`str` or `os.PathLike`):
The name of the folder to load all relevant weights and states.
model_stmodelsates (`List[torch.nn.Module]`):
models (`List[torch.nn.Module]`):
A list of model instances
optimizers (`List[torch.optim.Optimizer]`):
A list of optimizer instances
Expand Down

0 comments on commit fa17f20

Please sign in to comment.