Stand alone self attention combine with CycleGAN

leaderj1001 / Stand-Alone-Self-Attention

Implementing Stand-Alone Self-Attention in Vision Models using Pytorch

MIT License

456 stars 83 forks source link

Stand alone self attention combine with CycleGAN #26

Open Jingtianci opened 3 years ago

Jingtianci commented 3 years ago

My task is MRI-based pseudo-CT generation. I combine the SASA with the CycleGAN(https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix). Considering the GPU memory usage I halved the number of feature maps. I found that the network without SASA performs best. When I add SASA to D, the results are really bad. Did anyone manage to add SASA to CycleGAN? Hoping for a reply, Thank you.