Views
No views yet
lapvqa-pretrain-captioning).
Checkpoint is a plain DiffVQAHead state dict (vis_dim=1024).| BLEU-4 | ROUGE-2 | RadGraph-s | BERTScore F1 |
|---|---|---|---|
| 0.468 | 0.562 | 0.303 | 0.938 |
1import torch
2from lapvqa.diffvqa.model import DiffVQAHead
3
4ckpt = torch.load("pretrain-captioning_best.pt", map_location="cpu")
5head = DiffVQAHead(vis_dim=1024)
6head.load_state_dict(ckpt)
7head.eval()
8# pair with encoder_final.pt from lapvqa-pretrain-captioning