![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习框架pytorch
惟江上清风
东方欲晓,莫道君行早,踏遍青山人未老,风景这边独好!
展开
-
pytorch中网络loss传播和参数更新理解
相比于2018年,在ICLR2019提交论文中,提及不同框架的论文数量发生了极大变化,网友发现,提及tensorflow的论文数量从2018年的228篇略微提升到了266篇,keras从42提升到56,但是pytorch的数量从87篇提升到了252篇。TensorFlow: 228--->266Keras: 42--->56Pytorch: 87--->252在使...原创 2018-12-26 21:46:02 · 14714 阅读 · 3 评论 -
pytorch中自定义损失函数
在使用深度学习中,大家有各种各样的训练任务,torch.optim中存在的优化算法难免不能满足大家的需求,此时就需要自定义损失函数了,可参考如下过程:import torch.nn as nnclass DiceLoss(nn.Module): def __init__(self): super(DiceLoss, self).__init__() ...原创 2018-12-26 22:11:11 · 3068 阅读 · 3 评论 -
pytorch中数据集的划分方法及eError: take(): argument 'index' (position 1) must be Tensor, not numpy.ndarray错误原因
在使用pytorch框架时,难免需要对数据集进行训练集和验证集的划分,一般使用sklearn.model_selection中的train_test_split方法该方法使用如下:from sklearn.model_selection import train_test_splitimport numpy as npimport torchimport torch.autogra...原创 2018-12-27 21:55:14 · 20850 阅读 · 3 评论 -
ubuntu下conda虚拟环境的操作,cuda,cudnn版本的查询, pytorch的安装
一. ubuntu下conda虚拟环境的操作随着深度学习的发展,tensorflow, keras, pytorch等深度学习框架的兴起和发展,或者多用户的使用情况,使得在ubuntu下我们可能需要安装多个深度学习框架,为了避免各个深度学习框架之间的影响,可以使用conda创建不同的虚拟环境,方便使用。在conda下操作虚拟环境有以下几个重要的命令:(1)查看conda的虚拟环境有哪些...原创 2019-01-10 12:00:43 · 16190 阅读 · 4 评论 -
ubuntu18.04下安装cuda, cudnn,Anaconda3(python3), pytorch, 显卡gtx 1080Ti
一. 更新显卡驱动: 参考博客:https://blog.csdn.net/gdymind/article/details/82153643 可以用nvidia-smi命令进行查看:nvidia-smi二. 安装cuda10,cudnn,可以参考博客:https://blog.csdn.net/qq_32408773/article/details/84112166,...原创 2019-02-26 09:12:51 · 655 阅读 · 0 评论 -
pytorch多GPU训练和测试的问题: RuntimeError: Error(s) in loading state_dict for CPN:
Windows10上,使用两块显卡,训练pytorch模型,语句如下:model = network()if torch.cuda.device_count() > 1: model = nn.DataParallel(model) # 多gpu上训练model.cuda()然后在测试时,导入模型会报如下类似错误:RuntimeError: Error(s) ...原创 2019-03-15 14:55:30 · 3998 阅读 · 1 评论