GAN实现笔记

devnag代码阅读

github链接:https://github.com/devnag/pytorch-generative-adversarial-networks

torch.nn:神经网络(neural network)工具箱
torch.optim:实现了多种优化算法的包
torch.autograd:自动求导模块,将tensor变量包含进variable对象中即可。
torch.nn.Module:包含网络各层的定义及forward方法

0%