| 24 | # for model backward compatibility |
| 25 | # previously model was wrapped by DataParallel module |
| 26 | class DummyModule(nn.Module): |
| 27 | def __init__(self, model): |
| 28 | super(DummyModule, self).__init__() |
| 29 | self.module = model |
| 30 | |
| 31 | def forward(self, *xs, **kwargs): |
| 32 | return self.module(*xs, **kwargs) |
| 33 | |
| 34 | class NetworkFactory(object): |
| 35 | def __init__(self, db): |