# Codeblock 7
class CBAM(nn.Module):
    def __init__(self, num_channels):    #(1)
        super().__init__()
        
        self.cam = CAM(num_channels=num_channels)
        self.sam = SAM()
        
    def forward(self, x):
        print(f'original\t\t: {x.size()}')
        
        x = self.cam(x)
        print(f'after cam\t\t: {x.size()}')
        
        x = self.sam(x)
        print(f'after sam\t\t: {x.size()}')
        
        return x