(self, backbone_module, neck_module, decoder_module)
| 28 | |
| 29 | class aio_entry(nn.Module): |
| 30 | def __init__(self, backbone_module, neck_module, decoder_module): |
| 31 | super(aio_entry, self).__init__() |
| 32 | self.backbone_module = backbone_module |
| 33 | self.neck_module = neck_module |
| 34 | self.decoder_module = decoder_module |
| 35 | if self.decoder_module.task == 'ssl': |
| 36 | self.m = 0.99 |
| 37 | add_task_specific(self, False) |
| 38 | add_aio_backbone_specific(self.backbone_module, True, self.backbone_module.task_sp_list) |
| 39 | add_aio_neck_specific(self.neck_module, True, self.neck_module.task_sp_list) |
| 40 | add_aio_decoder_specific(self.decoder_module, True, self.decoder_module.task_sp_list, |
| 41 | self.decoder_module.neck_sp_list) |
| 42 | |
| 43 | def forward(self, input_var, current_step): |
| 44 | if self.decoder_module.task != 'ssl': |
no test coverage detected