Update README.md

README.md

@@ -23,9 +23,20 @@ split: 'late'         # represents the late fusion strategy
 ### Load Model
 
 ```
+# Load backnone
 model = RadNet(num_cls=num_classes, backbone=backbone, depth=depth, ltype=ltype, augment=augment, fuse=fuse, ke=ke, encoded=encoded, adapter=adapter)
 pretrained_weights = torch.load("path/to/pytorch_model_32_late.bin")
 missing, unexpect = model.load_state_dict(pretrained_weights,strict=False)
+print("missing_cpt:", missing)
+print("unexpect_cpt:", unexpect)
+
+# If KE is set True, load text encoder
+medcpt = MedCPT_clinical(bert_model_name = 'ncbi/MedCPT-Query-Encoder')
+checkpoint = torch.load('path/to/epoch_state.pt',map_location='cpu')['state_dict']
+load_checkpoint = {key.replace('module.', ''): value for key, value in checkpoint.items()}
+missing, unexpect = medcpt.load_state_dict(load_checkpoint, strict=False)
+print("missing_cpt:", missing)
+print("unexpect_cpt:", unexpect)
 ```
 
 ## Why we provide this checkpoint?