
机器学习
zzw小凡
这个作者很懒,什么都没留下…
展开
-
xgboost 在 win7 下安装
首先下载mingw 下载xgboost 安装xgboost 第三步转载自httpwwwcnblogscomcutdp5497929html 1.首先下载mingw:https://sourceforge.net/projects/mingw-w64/postdownload?source=dlp 安装时选择64位 其余不变2.下载xgboosthttps://github.com/dmlc/xg原创 2016-10-12 15:35:57 · 419 阅读 · 0 评论 -
pytorch 使用预训练层
pytorch 使用预训练层将其他地方训练好的网络,用到新的网络里面pytorch 使用预训练层 加载预训练网络 加载新网络 更新新网络参数加载预训练网络1.原先已经训练好一个网络 AutoEncoder_FC() 2.首先加载该网络,读取其存储的参数 3.设置一个参数集cnnpre = AutoEncoder_FC() cnnpre.load_state_dict(torch.load('au原创 2017-11-23 09:45:13 · 1662 阅读 · 1 评论 -
Pytorch Multi GPU 使用
1 检测torch 是否支持gpu import torch print(torch.cuda.version) #查看cuda版本 print(torch.cuda.is_available()) #查看pytorch 是否能用 cuda --9.0.176 --True 如果输出为true 就为可用,如果只有版本号,且输出为False,则需要重新装一下显卡的驱动 2 前向计算 device...原创 2019-06-20 16:28:38 · 1127 阅读 · 0 评论