tensorflow2
couldn
这个作者很懒,什么都没留下…
展开
-
could not load cublas64_11.dll
在文件目录下有该文件,环境变量也配好了,但还是报错could not load,这时候就得。原创 2022-12-02 18:29:23 · 3367 阅读 · 1 评论 -
tf.data dataset filter筛选数据,解决报错:ValueError: `predicate` return type must be convertible to a scalar
官网针对这个函数的讲解十分简陋,并不能用于实际,官网教程:tf.data.TFRecordDataset | TensorFlow Core v2.9.1https://tensorflow.google.cn/api_docs/python/tf/data/TFRecordDataset?hl=en 下面说说我的实现,废话不多说,直接上代码,看代码是最清晰的:注释的写法也是可以的。接下来说下我解决这个问题的过程:写成:报错根本这个报错信息我反复找如何把tensorspec转化为sca原创 2022-06-30 15:42:50 · 1245 阅读 · 0 评论 -
实现batch内负采样和解决报错:tensorflow.python.framework.errors_impl.InvalidArgumentError
在我们实现双塔等模型时一些负样本难以获取,这时我们可以通过batch内负采样的方式来实现训练,此时采集的样本数据仅需要正样本就够了。我们先来看一个案例,通过此案例来理解此代码运行这段代码的输出是:此时我们将我们的代码用于我们的模型之中,然后加上softmax,loss改为category_loss代码就可运行一半了,但是我们发现仿佛在一个epoch快要运行结束时报错了,报错为:它的原因为:我们的数据被分为一个batch_size,一个batch_size的输入模型中,在最后总会剩余不足一个原创 2022-06-20 16:14:52 · 2725 阅读 · 1 评论 -
linux下安装cuda11.2(从英伟达官网下载安装)
wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run再执行sudo sh cuda_11.2.0_460.27.04_linux.run原创 2021-11-10 12:11:15 · 2214 阅读 · 0 评论 -
tensorflow2 多目标模型子类化实现
直接上代码,import datetimeimport numpy as npimport pandas as pdfrom matplotlib import pyplot as pltimport tensorflow as tffrom tensorflow.keras import layers, models, Model# ============================================================================原创 2021-07-09 09:47:13 · 308 阅读 · 0 评论 -
tensorboard2使用指南。
from tensorflow.keras.callbacks import TensorBoardtbCallBack = TensorBoard(log_dir="./log", histogram_freq=1, write_grads=True)model.fit([x_train, x_train_masks], y_train, batch_size=batch_size, epochs=epochs, validation_split=0.2, callbacks=[tbCa.原创 2021-06-30 11:31:35 · 171 阅读 · 0 评论 -
training=true or false
目前的经验来看,training=false会使预测出来的值分布更加离散,差别相对较大;training=true预测出来的数值差别比较的小,甚至无差别。原创 2021-06-22 15:03:37 · 986 阅读 · 0 评论 -
package not found;安装烦人的cuda;安装烦人的cudnn
当我们安装包时可以使用conda install,也可以使用pip install,这两种没什么区别,但是conda install包的数量更多,更容易找到我们需要的包.但是就是安装这个cudatoolkit11.0,conda install无能为力。此时我们运行conda install -c conda-forge cudatoolkit==11.0就可以了,这只是将数据源改为了conda-forge这个channel,很多包原本没有的使用这个channel就可以。比如conda sear原创 2021-01-19 18:43:10 · 1744 阅读 · 0 评论