Commit
·
1a22dd9
1
Parent(s):
9ba400e
Update README.md
Browse files
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:
|