(self, channel=512,reduction=16,kernel_size=7, use_cls_token=False)
| 85 | |
| 86 | class CBAMBlock(nn.Module): |
| 87 | def __init__(self, channel=512,reduction=16,kernel_size=7, use_cls_token=False): |
| 88 | super().__init__() |
| 89 | self.ca = ChannelAttention(channel=channel,reduction=reduction) |
| 90 | self.sa = SpatialAttention(kernel_size=kernel_size) |
| 91 | self.use_cls_token = use_cls_token |
| 92 | |
| 93 | |
| 94 | def init_weights(self): |
no test coverage detected