(self, backbone_module, neck_module, decoder_module)
| 72 | |
| 73 | class backbone_aio_entry(nn.Module): |
| 74 | def __init__(self, backbone_module, neck_module, decoder_module): |
| 75 | super(backbone_aio_entry, self).__init__() |
| 76 | self.backbone_module = backbone_module |
| 77 | self.neck_module = neck_module |
| 78 | self.decoder_module = decoder_module |
| 79 | add_task_specific(self, False) |
| 80 | add_aio_backbone_specific(self.backbone_module, True, self.backbone_module.task_sp_list) |
| 81 | add_neck_specific(self.neck_module, True) |
| 82 | add_decoder_specific(self.decoder_module, True) |
| 83 | |
| 84 | def forward(self, input_var, current_step): |
| 85 | x = self.backbone_module(input_var) # {'image': img_mask, 'label': target_mask, 'filename': img_name, 'backbone_output':xxx} |
nothing calls this directly
no test coverage detected