QiaoyuZheng commited on
Commit
1a22dd9
·
1 Parent(s): 9ba400e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -0
README.md CHANGED
@@ -20,6 +20,14 @@ augment: True # represents the medical data augmentation
20
  split: 'late' # represents the late fusion strategy
21
  ```
22
 
 
 
 
 
 
 
 
 
23
  ## Why we provide this checkpoint?
24
 
25
  All the early fusion checkpoint can be further finetuned from this checkpoint. If you need other checkpoints using different parameter settings, there are two possible ways:
 
20
  split: 'late' # represents the late fusion strategy
21
  ```
22
 
23
+ ### Load Model
24
+
25
+ ```
26
+ model = RadNet(num_cls=num_classes, backbone=backbone, depth=depth, ltype=ltype, augment=augment, fuse=fuse, ke=ke, encoded=encoded, adapter=adapter)
27
+ pretrained_weights = torch.load("path/to/pytorch_model_32_late.bin")
28
+ missing, unexpect = model.load_state_dict(pretrained_weights,strict=False)
29
+ ```
30
+
31
  ## Why we provide this checkpoint?
32
 
33
  All the early fusion checkpoint can be further finetuned from this checkpoint. If you need other checkpoints using different parameter settings, there are two possible ways: