diff --git a/deepmatcher/models/modules.py b/deepmatcher/models/modules.py index b59e526..eb92510 100644 --- a/deepmatcher/models/modules.py +++ b/deepmatcher/models/modules.py @@ -234,7 +234,7 @@ def forward(self, x1, x2): """ def __getitem__(self, name): - return getattr(self, name) + return self._modules[name] def __setitem__(self, name, module): setattr(self, name, module)