
pytorch
prismformore
这个作者很懒,什么都没留下…
展开
-
基于PyTorch的GAN框架TorchGAN:用架构级API轻松定制GAN项目
https://mp.weixin.qq.com/s?__biz=MzA3MzI4MjgzMw==&mid=2650754528&idx=5&sn=af9172c63da653dd8f6bec18c6e223f4&chksm=871a899eb06d008802709464f205d725d09935a118df797778dff28f3fd5a4907c81bd8...转载 2018-12-26 09:48:46 · 204 阅读 · 0 评论 -
Pytorch初始化网络参数
单层 To initialize the weights of a single layer, use a function from torch.nn.init. For instance: conv1 = torch.nn.Conv2d(…) torch.nn.init.xavier_uniform(conv1.weight) Alternatively, you can modify the...转载 2019-01-09 11:44:57 · 2753 阅读 · 0 评论 -
Pytorch求Hessian矩阵及二阶导的方法
可以对梯度再次求导,用来惩罚梯度变化过快的情况。 也可以用来算Hessian矩阵。 官方举例说明: import torch from torchvision.models import resnet18 from torch.autograd import Variable model = resnet18().cuda() # dummy inputs for the example in...原创 2019-05-16 21:57:33 · 7799 阅读 · 2 评论