We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1358b72 commit aff62c9Copy full SHA for aff62c9
1 file changed
ignite/distributed/comp_models/native.py
@@ -149,8 +149,13 @@ def _create_from_backend(
149
init_pg_kwargs["world_size"] = int(os.environ["WORLD_SIZE"]) if world_size is None else world_size
150
init_pg_kwargs["rank"] = int(os.environ["RANK"]) if rank is None else rank
151
152
- # forward any remaining kwargs to support custom arguments like pg_options, etc.
153
- init_pg_kwargs.update(kwargs)
+ # Only forward kwargs that are valid parameters of dist.init_process_group
+ import inspect
154
+
155
+ sig = inspect.signature(dist.init_process_group)
156
+ for key, val in kwargs.items():
157
+ if key in sig.parameters:
158
+ init_pg_kwargs[key] = val
159
160
dist.init_process_group(backend, init_method=init_method, **init_pg_kwargs)
161
0 commit comments