12 2019 档案

摘要:Variable(adv_img_tea).cuda() 变量后面.cuda() 阅读全文
posted @ 2019-12-11 18:58 闪存第一菜鸡 阅读(4227) 评论(0) 推荐(0)
摘要:1. 出错代码行 计算交叉熵是出现异常提示:RuntimeError: multi-target not supported at /opt/conda/conda-bld/pytorch_1549635019666/work/aten/src/THNN/generic/ClassNLLCriter 阅读全文
posted @ 2019-12-10 20:11 闪存第一菜鸡 阅读(16846) 评论(0) 推荐(1)
摘要:解决方法: Pytorch使用Pickle来处理保存/加载模型,这个问题实际上是Pickle的问题,而不是Pytorch。 解决方法也非常简单,只需显式地导入类定义。即将包含类定义的文件复制粘贴到与要运行的文件同一文件夹下,再 import Class! 阅读全文
posted @ 2019-12-10 18:19 闪存第一菜鸡 阅读(9188) 评论(0) 推荐(0)
摘要:pytorch的模型和参数是分开的,可以分别保存或加载模型和参数。 pytorch有两种模型保存方式:一、保存整个神经网络的的结构信息和模型参数信息,save的对象是网络net 二、只保存神经网络的训练模型参数,save的对象是net.state_dict() 对应两种保存模型的方式,pytorch 阅读全文
posted @ 2019-12-10 11:18 闪存第一菜鸡 阅读(1845) 评论(0) 推荐(0)
摘要:多进程导致的内存不够用了吧 将数据集读取的进程修改为 trainset = torchvision.datasets.CIFAR100(root=args.data_path, train=True, download=False, transform=transform_train) trainl 阅读全文
posted @ 2019-12-08 18:25 闪存第一菜鸡 阅读(14103) 评论(0) 推荐(0)
摘要:pytorch 加载数据时报错: RuntimeError: An attempt has been made to start a new process before the current process has finished its bootstrapping phase. This p 阅读全文
posted @ 2019-12-08 17:24 闪存第一菜鸡 阅读(644) 评论(0) 推荐(0)
摘要:这个错误一般是卷积和全连接直接对接的时候,大小不匹配导致的。 可以查看卷积结束后 图片大小 ,然后对应修改全连接的输入大小即可。 def forward(self,x): out = self.conv(x) out = out.view(out.size(0),-1) # print("-->{} 阅读全文
posted @ 2019-12-07 16:44 闪存第一菜鸡 阅读(5777) 评论(0) 推荐(0)
摘要:xshell 和xftp 个人免费版 https://www.netsarang.com/download/down_result.html 阅读全文
posted @ 2019-12-05 21:57 闪存第一菜鸡 阅读(133) 评论(0) 推荐(0)
摘要:报错原因分析: train_loss += loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错,总的来说是版本更新问题. 解决方法: 将datra[0]换成item() 即可 阅读全文
posted @ 2019-12-04 21:31 闪存第一菜鸡 阅读(2014) 评论(0) 推荐(0)