[nfc] fix typo colossalai/zero (#3923)

This commit is contained in:
digger yu
2023-06-08 00:01:29 +08:00
committed by GitHub
parent a9d1cadc49
commit de0d7df33f
6 changed files with 11 additions and 11 deletions

View File

@@ -238,7 +238,7 @@ def initialize(model: nn.Module,
loaded into gpc.config.
Args:
model (:class:`torch.nn.Module` or Callbale): Your model instance or a function to build the model.
model (:class:`torch.nn.Module` or Callable): Your model instance or a function to build the model.
optimizer (:class:`torch.optim.optimizer.Optimizer` or :class:`Type[torch.optim.optimizer]`):
Your optimizer instance.
criterion (:class:`torch.nn.modules.loss._Loss`, optional): Your criterion instance.