How can I integrate a couple of self-attention modules to the resnet architecture? Could you please help me in breaking down this task? Or if there is already a code-base you could suggest me to have a look at?