如何在TensorFlow中高效地对张量的切片进行赋值
我希望在TensorFlow 2.x中的一个模型中，将输入张量的某些切片分配给一些值（我正在使用2.2版本，但接受解决方案为2.1）。 我尝试的非工作模板如下： import tensorflow as tf from tensorflow.keras.models import Model...
tf.model.fit()中的batch_size与tf.data.Dataset中的batch_size有何不同？
我拥有一个很大的数据集，可以存储在主机内存中。然而，当我使用tf.keras训练模型时，会导致GPU内存不足的问题。于是我研究了tf.data.Dataset，并想要使用其batch()方法来批处理训练集，以便可以在GPU上执行model.fit()。根据文档，一个示例如下： train_d...
Tensorflow 2.1.0 - 函数构建代码之外的操作正在传递一个“Graph”张量
我正在尝试实现一篇最近的论文。这个实现的一部分涉及从tf 1.14转移到tf 2.1.0。代码在tf 1.14下工作正常，但现在不再工作。 注意：如果我禁用急切执行tf.compat.v1.disable_eager_execution()，那么代码就会按预期工作。 这是解决方案吗？我...
TensorFlow的fit方法报错：TypeError: Cannot clone object error
我正在使用基本的CNN模型来对我的数据进行分类。我的输入数据的尺寸是（325，20，244，244）。我所使用的代码如下： model = Sequential() model.add(Dense(2, activation='relu', input_shape=X_train.shape...
将model.fit_generator转换为model.fit
我有以下的代码： train_datagen = ImageDataGenerator( rescale=1./255, shear_range=0.2, zoom_range=0.2, horizontal_flip=True...
如何在Tensorflow 2.0中获取除准确度以外的其他指标？
我是Tensorflow世界的新手，正在处理mnist数据集分类的简单示例。我想知道除了准确度和损失之外（并可能展示它们），如何获取其他指标（例如精确率、召回率等）。这是我的代码： from __future__ import absolute_import, division, print...
如何在tensorflow2-gpu中调试内存不足问题
我正在使用带有tf.data.Dataset的tensorflow-2 gpu进行训练。 当使用小模型进行训练时，一切正常。 当训练一个更大的模型时，一开始一切顺利：GPU被使用，在第一个时期中没有问题（但我使用了大部分GPU内存）。 在验证期间，我遇到了CUDA_ERROR_OUT_OF...
什么是用于CNN的numpy版TensorFlow Xavier初始化器的等效物？
我希望能够使用基本函数在NumPy中重新创建Xavier初始化，就像TensorFlow2在CNN中所做的那样。以下是我学习如何在NumPy中进行Xavier初始化的方式： # weights.shape = (2,2) np.random.seed(0) nodes_in = 2*2 we...
Tensorflow 2.0无法使用GPU，cuDNN出了些问题吗？：无法获取卷积算法。这可能是因为cuDNN初始化失败。
我正在尝试理解和调试我的代码。 我尝试在GPU上使用tf2.0 / tf.keras开发的CNN模型进行预测，但出现以下错误消息。 有人能帮助我修复吗？ 这是我的环境配置： enviroments: python 3.6.8 tensorflow-gpu 2.0.0-rc0 nvidia ...
Tensorflow在GPU上比CPU慢
使用Tensorflow后端的Keras，我正在尝试训练一个LSTM网络。但是在GPU上运行比在CPU上运行慢得多。我正在使用fit_generator函数来训练LSTM网络。CPU每个时代大约需要250秒，而GPU则需要大约900秒。我的GPU环境中包含以下软件包： keras-appli...