(self, backbone, args)
| 53 | |
| 54 | class CGFormer(nn.Module): |
| 55 | def __init__(self, backbone, args): |
| 56 | super(CGFormer, self).__init__() |
| 57 | self.backbone = backbone |
| 58 | self.decoder = Decoder(args) |
| 59 | self.text_encoder = BertModel.from_pretrained(args.bert) |
| 60 | self.text_encoder.pooler = None |
| 61 | |
| 62 | def forward(self, x, text, l_mask, mask=None): |
| 63 | input_shape = x.shape[-2:] |
nothing calls this directly
no test coverage detected