Skip to content

Commit 603ec6c

Browse files
authored
Update README.md (#508)
1 parent 996575e commit 603ec6c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ You need to choose the target layer to compute CAM for.
101101
Some common choices are:
102102
- FasterRCNN: model.backbone
103103
- Resnet18 and 50: model.layer4[-1]
104-
- VGG and densenet161: model.features[-1]
104+
- VGG, densenet161 and mobilenet: model.features[-1]
105105
- mnasnet1_0: model.layers[-1]
106106
- ViT: model.blocks[-1].norm1
107107
- SwinT: model.layers[-1].blocks[-1].norm1

0 commit comments

Comments
 (0)