tensorflow
文章平均质量分 94
momaojia
NLP菜鸡
展开
-
centos7下编译tensorflow1.15 GPU版本(cuda11)
因为最近要把cuda10.0升级到cuda11,但是cuda11都不支持tensorflow1.1x版本,所以尝试从源码编译,目前tensorflow官方给出了ubuntu版本的编译方法,nvidia也提供了一款基于ubuntu的nvidia-tensorflow,直接安装在centos上,期间出现了各种问题,因此本文记录了在centos上编译tensorflow-gpu1.15版本的过程,仅供大家参考。 下表展示了tensoflow和cuda、gcc之间的对应关系[外链图片转存失原创 2021-11-24 21:15:25 · 4821 阅读 · 3 评论 -
tensorflow实现给bert-crf不同层设置不同的学习率
前段时间看了苏神的文章,关于CRF层可能存在学习不充分问题,文章地址:https://kexue.fm/archives/7196,简而言之就是bert拟合能力太强,可以迅速的学习出字的标签分布,但是CRF部分和BERT没什么关系,前面的字标签已经学习的很好了,导致转移的矩阵的梯度会非常小。想到本人前段时间正好做的实体识别任务,有时候会存在标签乱序的现象,故把以前的代码CRF层中加入不同的...原创 2020-03-22 18:24:47 · 2599 阅读 · 1 评论 -
Ubuntu16.04安装 tensorflow-gpu CUDA9.0+cudnn 7.0
本篇适合第一次安装tensorflow-gpu的人,折腾了好久,一开始装了CUDA 10.1,发现tensorflow 1.13版本调用不起来GPU,后来看了官方的说明,CUDA10.1好像要2.0以上的tensorflow,所以放弃了CUDA10.1,装起了CUDA 9.0,以下为安装步骤一、安装nvidia驱动首先在终端 输入nvidia-smi 如果有以下显示,说明驱动正常...原创 2020-02-22 10:31:09 · 451 阅读 · 2 评论 -
tensorflow 中的embedding 报错问题解决
今天TensorFlow中的tf.contrib.layers.embed_sequence来对输入进行embedding时候,发现报了如下的错误: InvalidArgumentError (see above for traceback): indices[1,2] = 6 is not in [0, 6) [[Node: EmbedSequence_8/embedding_lo原创 2017-08-24 17:42:34 · 11512 阅读 · 3 评论 -
tensorflow 中tf.concat()用法
concat()是将tensor沿着指定维度连接起来。其中tensorflow1.3版中是这样定义的:concat(values,axis,name='concat') 一、对于2维来说,0表示行,1表示列t1 = [[1, 2, 3], [4, 5, 6]]t2 = [[7, 8, 9], [10, 11, 12]]with tf.Session() as sess原创 2017-08-26 17:24:07 · 10784 阅读 · 0 评论 -
基于简单seq to seq 的聊天机器人+代码实现 (tensorfow 1.1版本)
一、seqto seq模型简介基本Encoder-Decoder模型输入的序列为['A','B', 'C', ''],输出序列为['W','X', 'Y', 'Z', '']二、tensorflow1.1seq2seq接口自从tensoflow1.0.0开始,开始出现新的seq2seq接口。新的接口主要采用动态展开,旧接口是动态展开的。静态展开(static原创 2017-09-04 15:01:21 · 5052 阅读 · 1 评论 -
java 调用训练好的keras模型,tensorflow Pb模型
这篇文章主要给初次想使用java 调用keras或者使用tensorflow训练好的模型,下面将详细介绍环境安装以及调用步骤。假设你忘记了如何配置java环境,可以从第一部分看起,否则直接转入第二部分。一、首先配置java环境1、下载JDK1.8版本,并且安装2、配置环境变量,本人安装路径为默认路径C:\ProgramFiles\Java\jdk1.8.0_1313、在path里输入%HOME_J...原创 2018-05-01 18:27:08 · 14551 阅读 · 4 评论