tensorflow
pan15125284
这个作者很懒,什么都没留下…
展开
-
tensorflow在训练模型的时候如何指定GPU进行训练
2.通过CUDA_VISIBLE_DEVICES来指定.同样使用第2块GPU来训练模型,我们可以在我们的python代码中加入:import osos.environ['CUDA_VISIBLE_DEVICES']='2'实验效果如下:我们可以看到,在训练模型的时候,使用了第2块GPU,并且其它几块GPU也没有被占,这种就相当于在我们运行程序的时候,将除第2块以外的GPU全部屏...原创 2020-02-20 14:55:25 · 954 阅读 · 0 评论 -
关于keras构建的神经网络的输出结果解释
ETA:Estimated Time of Arrival。Loss:系统的损失。acc:Accuracy正确率。原创 2020-02-20 14:44:50 · 775 阅读 · 0 评论 -
python 安装包问题
一、No module named 'keras.contrib'1、下载https://github.com/keras-team/keras-contrib2、解压3、打开控制面板,进入到解压目录下(G:\developInstall\...\keras-contrib-master)4、执行 python setup.py install...原创 2019-12-31 09:33:03 · 190 阅读 · 0 评论 -
layers.Embedding
input:(input_dim,output_dim,input_length=None)output:(input_length,output_dim)import tensorflow as tffrom tensorflow.keras import layersmodel = tf.keras.Sequential()model.add(layers.Embed...原创 2019-11-01 13:41:48 · 2289 阅读 · 1 评论 -
NLP词嵌入Word embedding实战项目
1. 将文本表示为数字机器学习模型以向量(数字数组)作为输入,在处理文本时,我们必须首先想出一个策略,将字符串转换为数字(或将文本“向量化”),然后再将其提供给模型。在本节中,我们将研究三种策略。1.1. 独热编码(One-hot encodings)首先,我们可以用“one-hot”对词汇的每个单词进行编码,想想“the cat sat on the mat”这句话,这个句子中的词汇...原创 2019-06-24 15:03:01 · 924 阅读 · 0 评论 -
tensorflow2.0 保存和加载模型
1、保存全模型model.save('the_save_model.h5')new_model = keras.models.load_model('the_save_model.h5')2、保持为SavedModel文件keras.experimental.export_saved_model(model, 'saved_model')new_model = keras.ex...原创 2019-06-25 12:23:10 · 12782 阅读 · 0 评论 -
tensorflow2.0 numpy.ndarray 与tenor直接互转
import numpy as npimport tensorflow as tfa = np.random.random((5,3))b = np.random.randint(0,9,(3,1))c = tf.tensordot(a.astype(np.float),b.astype(np.float),axes=1)# tensor 转ndarraydn = c.numpy(...原创 2019-06-25 08:59:16 · 9289 阅读 · 6 评论 -
tensorflow tf.argmax() 用法 例子
argmax()官方文档如下:tf.argmax(input, dimension, name=None)Returns the index with the largest value across dimensions of a tensor.Args:input: A Tensor. Must be one of the following types: float32, f...原创 2019-06-25 08:37:31 · 607 阅读 · 0 评论 -
安装TensorFlow2.0 Cannot uninstall 'wrapt'.
1、原始版本import tensorflow as tfprint(tf.__version__)1.13.12、安装TensorFlow2.0pip install tensorflow==2.0.0-beta02.1 Cannot uninstall 'wrapt'.解决:pip install -U --ignore-installed wrapt enu...原创 2019-06-10 09:02:56 · 3070 阅读 · 0 评论 -
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
步骤一:在PATH中加入cudasudo vim ~/.bashrc在末尾加入:export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}export CUDA_HOME=/usr/...原创 2019-06-05 14:23:22 · 837 阅读 · 0 评论