Unverified Commit 308f1057 authored by Mehdi Mirzazadeh's avatar Mehdi Mirzazadeh Committed by GitHub
Browse files

use ModuleLst for submodules (#722)

parent 1cc4c837
......@@ -20,7 +20,7 @@ class MultiInputSequential(nn.Module):
def __init__(self, *modules: nn.Module) -> None:
super().__init__()
self.modules_list = modules
self.modules_list = nn.ModuleList(modules)
def forward(self, *inputs: Tuple[Tensor]) -> Tensor: # type: ignore
input = self.modules_list[0](*inputs)
......
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