tensorflow
pan15125284
这个作者很懒,什么都没留下…
展开
-
tensorflow在训练模型的时候如何指定GPU进行训练
2.通过CUDA_VISIBLE_DEVICES来指定. 同样使用第2块GPU来训练模型,我们可以在我们的python代码中加入: import os os.environ['CUDA_VISIBLE_DEVICES']='2' 实验效果如下: 我们可以看到,在训练模型的时候,使用了第2块GPU,并且其它几块GPU也没有被占,这种就相当于在我们运行程序的时候,将除第2块以外的GPU全部屏...原创 2020-02-20 14:55:25 · 956 阅读 · 0 评论 -
关于keras构建的神经网络的输出结果解释
ETA:Estimated Time of Arrival。Loss:系统的损失。acc:Accuracy正确率。原创 2020-02-20 14:44:50 · 776 阅读 · 0 评论 -
python 安装包问题
一、No module named 'keras.contrib' 1、下载https://github.com/keras-team/keras-contrib 2、解压 3、打开控制面板,进入到解压目录下(G:\developInstall\...\keras-contrib-master) 4、执行 python setup.py install ...原创 2019-12-31 09:33:03 · 191 阅读 · 0 评论 -
layers.Embedding
input:(input_dim,output_dim,input_length=None) output:(input_length,output_dim) import tensorflow as tf from tensorflow.keras import layers model = 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 · 925 阅读 · 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 · 12786 阅读 · 0 评论 -
tensorflow2.0 numpy.ndarray 与tenor直接互转
import numpy as np import tensorflow as tf a = 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 转ndarray dn = c.numpy(...原创 2019-06-25 08:59:16 · 9291 阅读 · 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 · 608 阅读 · 0 评论 -
安装TensorFlow2.0 Cannot uninstall 'wrapt'.
1、原始版本 import tensorflow as tf print(tf.__version__) 1.13.1 2、安装TensorFlow2.0 pip install tensorflow==2.0.0-beta0 2.1 Cannot uninstall 'wrapt'. 解决: pip install -U --ignore-installed wrapt enu...原创 2019-06-10 09:02:56 · 3071 阅读 · 0 评论 -
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
步骤一:在PATH中加入cuda sudo 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 · 838 阅读 · 0 评论