| """ | |
| Contains an implementation of a Module that sequentially applies multiple models. | |
| Author: Ole-Christian Galbo Engstrøm | |
| E-mail: ocge@foss.dk | |
| """ | |
| import torch.nn as nn | |
| class CompoundModel(nn.Module): | |
| def __init__(self, *models): | |
| super(CompoundModel, self).__init__() | |
| self.models = nn.ModuleList(models) | |
| def forward(self, x): | |
| for model in self.models: | |
| x = model(x) | |
| return x | |