Skip to content

Commit ee39314

Browse files
authored
fix for config runner compatibility with DDP training (#1383)
1 parent 1c6cfdb commit ee39314

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

catalyst/runners/config.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
import logging
55
import os
66

7-
from torch import nn
87
from torch.utils.data import DataLoader, Dataset
98

109
from catalyst.callbacks import CheckpointCallback, ICheckpointCallback
@@ -287,7 +286,6 @@ def _get_model_from_params(**params) -> RunnerModel:
287286
model_key: ConfigRunner._get_model_from_params(**model_params)
288287
for model_key, model_params in params.items()
289288
}
290-
model = nn.ModuleDict(model)
291289
else:
292290
model = REGISTRY.get_from_params(**params)
293291
return model

0 commit comments

Comments
 (0)