(self, args)
| 12 | |
| 13 | class BasicModel(nn.Module): |
| 14 | def __init__(self, args): |
| 15 | nn.Module.__init__(self) |
| 16 | |
| 17 | self.backbone = getattr(backbones, args['backbone'])(**args.get('backbone_args', {})) |
| 18 | self.decoder = getattr(decoders, args['decoder'])(**args.get('decoder_args', {})) |
| 19 | |
| 20 | def forward(self, data, *args, **kwargs): |
| 21 | return self.decoder(self.backbone(data), *args, **kwargs) |