- 博客(5)
- 收藏
- 关注
RSS订阅原创 TensorFlow 1.13.1 CUDNN_STATUS_ALLOC_FAILED 错误解决办法
为使用faceswap玩一下换脸，在win10+rtx2070平台上搭建TensorFlow-gpu 1.13.1环境。安装完毕后import tensorflow没有问题，但在运行时总是报Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED错误，上网多方查证后认为cuda、cudnn、TensorFlow的版本搭配没有问题，但就是反复重...
2019-04-18 21:45:04
1630
原创 会议截稿日期倒计时网站
近期看到一个AI领域会议截稿日期倒计时的网站：aideadlin.es ，觉得非常不错。于是fork作者的repo做了一个电力电子与电气传动领域顶会倒计时的网站：PEMC Conference deadlines，欢迎相关专业的同学关注、完善～其他专业的同学也可以fork从而做一个本专业的，只需要非常简单的前端知识就可以了。...
2018-12-25 14:31:09
6820
1
原创 win10+rtx2070+tensorflow-gpu-1.9.0环境搭建
最近实验室新到一台工作站，配备了两块rtx2070。但为方便使用其他科研软件，只能使用win10。为学习深度有趣 - 人工智能实战项目合集 ，需安装TensorFlow 1.9.0环境。win10下搭建深度学习环境首推基于Anaconda的办法，网上资料很多，在此不再赘述，谨将遇到的坑记录如下。
2018-12-23 23:07:59
2623
2
原创 基于循环神经网络搭建语言模型
利用 PyTorch，本项目实现了一个基于word embedding和GRU的语言模型（Language Model，以下简称LM）。其包括一个encoder层、一个GRU层和一个decoder层，embedding维度与GRU的hidden state维度均为1500，采用了自己搭建的带有Layer Normalization（LN）的GRU模块，运用了dropout、学习初始状态、锁定encoder与decoder参数、梯度裁剪等技巧来提升模型性能，最终在测试集上模型的perplexity (PP)值
2018-12-18 23:30:40
1137
原创 基于卷积神经网络进行图像分类
利用TensorFlow（TF），本项目实现了一个基于full convolution stack、inception v2 module等模块的图片分类网络，纵向共包含10层包含参数的层，采用自己搭建的卷积层。对于dset1与dset2两个数据集采用同样的网络结构，分别进行训练。最终在dset1验证集（约含900张图，下同）上的Top1分类准确率约为0.52，在dset2验证集上的Top1准确率约为0.63。在K80显卡的单核上，该模型的训练速度约为205张图每秒，最终所得模型的checkpoint约为4
2018-12-18 23:11:51
20592
6
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人