Skip to content

Commit b38884b

Browse files
JKSenthilfacebook-github-bot
authored andcommitted
update error message for invalid distributed strategy (#945)
Summary: Pull Request resolved: #945 Give examples of strings to use as strategies Reviewed By: diego-urgell Differential Revision: D65766329 fbshipit-source-id: e8c1182ef752816e52f1ff8a9d5e3ad500217f0b
1 parent 753f05a commit b38884b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchtnt/utils/prepare_module.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ def prepare_module(
314314
if strategy:
315315
if not isinstance(strategy, str) and not isinstance(strategy, Strategy):
316316
raise ValueError(
317-
f"Unknown strategy received: {strategy}. Expect either str or Strategy dataclass"
317+
f"Unknown strategy received: {strategy}. Expect either str (one of 'ddp', 'fsdp', or 'noop') or Strategy dataclass"
318318
)
319319

320320
if isinstance(strategy, str):

0 commit comments

Comments
 (0)