![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
love_data_scientist
这个作者很懒,什么都没留下…
展开
-
pytorch sklearn dataset load_breast_cancer 逻辑回归(二分类)
import matplotlib.pyplot as plt import torch from torch import nn from torch import optim from sklearn.datasets import load_breast_cancer data_set = load_breast_cancer() x = torch.from_numpy(data_set...原创 2019-07-29 17:13:16 · 2738 阅读 · 0 评论 -
pytorch构建自己数据集合
使用pytorch构建自己的训练和测试数据集合,涉及自己数据处理类,数据变为tensor,数据分割等,为后续的训练准备了训练数据和测试数据 import torch from torch.utils.data import DataLoader, Dataset,TensorDataset,random_split import sys class label_featureDataS...原创 2019-08-15 17:09:59 · 415 阅读 · 0 评论 -
服务器无root的权限下pytorch1.2 踩坑(可联网)
1.服务器只安装了cuda9.2 2.安装anaconda Anaconda3-5.2.0-Linux-x86_64.sh python3.6 shAnaconda3-5.2.0-Linux-x86_64.sh 增加清华源: https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/ 如果卸载重装执行conda instal...原创 2019-08-30 20:54:58 · 1743 阅读 · 0 评论 -
pytorch1.1 单机单GPU程序改为单机多GPU程序方案
最近在用pytorch训练图文相关性模型,图片特征使用resnet抽取,文案特征使用bert抽取,把这两个特征合并为一个特征送入浅层神经网络中。 数据量:1千万条。 机器配置:单机4块GPU 现状:单个GPU跑一轮需要15个小时左右。 pytorch1.1对单机多GPU支持的很好,直接一条命令解决: nn.DataParallel(model).cuda() 方案1: 直接有两个模型...原创 2019-09-04 19:23:34 · 1108 阅读 · 0 评论 -
复习-优化算法
在线学习算法:FTRL 下面程序来自https://www.kaggle.com/jiweiliu/ftrl-starter-code/code alpha = .005 # learning rate beta = 1. # smoothing parameter for adaptive learning rate L1 = 0.2 # L1 regu...原创 2019-09-16 12:56:47 · 211 阅读 · 0 评论