Skip to content

Commit fd78425

Browse files
Richard Wangfacebook-github-bot
authored andcommitted
wiki nit: AutoUnit make session optional string (#781)
Summary: Pull Request resolved: #781 nit for TorchTNT wiki: https://www.internalfb.com/intern/staticdocs/torchtnt/overview.html for n00bs like me, it’s likely much better to start with strategy=None {F1482904808} Reviewed By: galrotem Differential Revision: D55947333 fbshipit-source-id: 238e7d7cd89968a424cbd0205473fb2aaa4f69cd
1 parent 1d26dfe commit fd78425

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/source/overview.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ The :class:`~torchtnt.framework.auto_unit.AutoUnit` implements the TrainUnit, Ev
144144
self,
145145
module: torch.nn.Module,
146146
device: torch.device,
147-
strategy: str,
147+
strategy: Optional[str],
148148
precision: Optional[str],
149149
gradient_accumulation_steps: int,
150150
*,

0 commit comments

Comments
 (0)