(feat1, feat2, pos1, pos2, shape1, shape2)
| 578 | feat1, feat2, pos1, pos2 = model._encode_image_pairs(img1, img2, shape1, shape2) |
| 579 | |
| 580 | def decoder(feat1, feat2, pos1, pos2, shape1, shape2): |
| 581 | dec1, dec2 = model._decoder(feat1, pos1, feat2, pos2) |
| 582 | with torch.cuda.amp.autocast(enabled=False): |
| 583 | res1 = model._downstream_head(1, [tok.float() for tok in dec1], shape1) |
| 584 | res2 = model._downstream_head(2, [tok.float() for tok in dec2], shape2) |
| 585 | return res1, res2 |
| 586 | |
| 587 | # decoder 1-2 |
| 588 | res11, res21 = decoder(feat1, feat2, pos1, pos2, shape1, shape2) |
no test coverage detected