fix for config runner compatibility with DDP training

bagxi · bagxi · commit 7b4b05aadb9e · 2021-12-21T11:43:28.000+03:00
diff --git a/catalyst/runners/config.py b/catalyst/runners/config.py
@@ -287,7 +287,6 @@ def _get_model_from_params(**params) -> RunnerModel:
                 model_key: ConfigRunner._get_model_from_params(**model_params)
                 for model_key, model_params in params.items()
             }
-            model = nn.ModuleDict(model)
         else:
             model = REGISTRY.get_from_params(**params)
         return model

Original file line number	Diff line number	Diff line change
`@@ -287,7 +287,6 @@ def _get_model_from_params(**params) -> RunnerModel:`
`287`	`287`	`model_key: ConfigRunner._get_model_from_params(**model_params)`
`288`	`288`	`for model_key, model_params in params.items()`
`289`	`289`	`}`
`290`		`- model = nn.ModuleDict(model)`
`291`	`290`	`else:`
`292`	`291`	`model = REGISTRY.get_from_params(**params)`
`293`	`292`	`return model`