Unverified Commit 1738d50c authored by Haian Huang(深度眸)'s avatar Haian Huang(深度眸) Committed by GitHub
Browse files

Use Sequential rather than nn.Sequential in build_model_from_cfg. (#992)

parent 0be04104
import torch.nn as nn from ..runner import Sequential
from ..utils import Registry, build_from_cfg from ..utils import Registry, build_from_cfg
...@@ -22,7 +21,7 @@ def build_model_from_cfg(cfg, registry, default_args=None): ...@@ -22,7 +21,7 @@ def build_model_from_cfg(cfg, registry, default_args=None):
modules = [ modules = [
build_from_cfg(cfg_, registry, default_args) for cfg_ in cfg build_from_cfg(cfg_, registry, default_args) for cfg_ in cfg
] ]
return nn.Sequential(*modules) return Sequential(*modules)
else: else:
return build_from_cfg(cfg, registry, default_args) return build_from_cfg(cfg, registry, default_args)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment