pytorch
竹篓有个天
如有侵权,请私信我
展开
-
python 代码调试问题 YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated RuntimeError
问题1:YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated, as the default Loader is unsa解决1:5.1弃用了load原本的用法在yaml.load(list, Loader=yaml.FullLoader)加上 Loader=yaml.FullLoader 就行了问题2:RuntimeError[Errno 32] Broken pipe:A...原创 2021-06-29 16:16:27 · 240 阅读 · 0 评论 -
(已解决)pytorch错误:RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED (安装cuda)
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED原因:pytorch与cuda版本不对先卸载了pytorch:conda uninstall pytorch,再安装cuda的话,会自动覆盖cuda版本。打开cmd,命令行输入import torchprint(torch.__version__)print(tor...原创 2020-02-08 17:54:29 · 54195 阅读 · 11 评论 -
pytorch实现wGAN(附代码)
WGAN在GAN上加入了wasserstein ditance做了改善:有一个梯度惩罚项,X是做了一个线性插值。梯度惩罚:惩罚系数取0.2,总训练5000次,批次为512,优化器同上。代码:# -*- coding: utf-8 -*-"""Created on Tue Jan 14 16:37:46 2020@author: ZM"""import torc...原创 2020-02-02 19:09:15 · 7474 阅读 · 12 评论 -
pytorch实现GAN(附代码)
GAN定义的生成器和判别器网络结构:GAN训练:首先训练D,每迭代训练5次D再迭代训练一次G,总训练5000次,批次为512。数据集为8个高斯混合模型,使得GAN去拟合一个圆上的8个分布点。根据GAN的loss函数,使得D尽可能的大,G尽可能的小,所以D取负数。过程中用到了visdom可视化工具。import visdomviz = visdom.Visdo...原创 2020-02-02 19:05:34 · 4970 阅读 · 1 评论