akanimax/attn_gan_pytorch

python package for self-attention gan implemented as extension of PyTorch nn.Module. paper -> https://arxiv.org/abs/1805.08318

语言：Python · GitHub 星标：19 · Fork：5

话题：artificial-intelligence · deep-learning · fagan · gan · sagan

在 GitHub 上查看原仓库