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 1c6cfdb commit ee39314Copy full SHA for ee39314
catalyst/runners/config.py
@@ -4,7 +4,6 @@
4
import logging
5
import os
6
7
-from torch import nn
8
from torch.utils.data import DataLoader, Dataset
9
10
from catalyst.callbacks import CheckpointCallback, ICheckpointCallback
@@ -287,7 +286,6 @@ def _get_model_from_params(**params) -> RunnerModel:
287
286
model_key: ConfigRunner._get_model_from_params(**model_params)
288
for model_key, model_params in params.items()
289
}
290
- model = nn.ModuleDict(model)
291
else:
292
model = REGISTRY.get_from_params(**params)
293
return model
0 commit comments