370得票16回答
如何防止TensorFlow分配整个GPU内存?

我工作的环境中计算资源是共享的,也就是说我们有几台服务器每台配备了几个 Nvidia Titan X GPU。 对于小到中等规模的模型,每张 Titan X 的 12 GB 内存通常足够让 2-3 人同时在同一块 GPU 上跑训练。如果模型足够小,单个模型无法充分利用 GPU 的所有计算单元,...

47得票5回答
Tensorboard在Jupyter中未被识别为魔术函数。

我想在Jupyter中使用最新的TensorFlow 2.0.0a0运行TensorBoard。 我已经安装了版本为1.13.1的TensorBoard和Python 3.6,但是当我尝试使用以下命令时:"%tensorboard --logdir {logs_base_dir}",出现错误提...

36得票2回答
自定义TensorFlow Keras优化器

假设我想编写一个符合 TensorFlow 版本>=2.0 的自定义优化器类,符合 tf.keras API。我对文档中所述的方法与实现中的方法感到困惑。 tf.keras.optimizers.Optimizer 的文档说明: ### Write a customized optim...

33得票3回答
我应该对所有函数使用@tf.function吗?

一份关于@tf.function的官方教程这样说: 使用tf.function将程序转换为图形以获得最佳性能并使您的模型可以在任何地方部署。由于AutoGraph的存在,令人惊讶的是,大量Python代码可以直接与tf.function配合使用,但仍然需要注意一些陷阱。 主要的收获和建议是...

19得票2回答
使用keras.utils.Sequence对象还是tf.data.Dataset来构建输入管道?

我目前正在使用tf.keras.utils.Sequence对象为CNN生成图像批次。我正在使用Tensorflow 2.2和Model.fit方法来训练模型。当我在tf.keras.model.fit(...)中设置use_multiprocessing=True时,在每个epoch中拟合模...

14得票2回答
在Tensorflow 2.1中,AttributeError: 'Tensor' object has no attribute 'numpy'。

我正在尝试在Tensorflow 2.1中转换Tensor的shape属性,但是我遇到了这个错误:AttributeError: 'Tensor' object has no attribute 'numpy' 我已经确认tf.executing_eagerly()的输出结果为True, 一...

11得票3回答
属性错误:模块“tensorflow_core.keras.layers.experimental.preprocessing”没有“RandomFlip”属性。

我使用 Tensorflow 2.1.0 版本。 在这段代码中 data_augmentation = tf.keras.Sequential([ tf.keras.layers.experimental.preprocessing.RandomFlip('horizontal'),...

10得票2回答
在Tensorflow中获取数据集的长度

source_dataset = tf.data.TextLineDataset('primary.csv') target_dataset = tf.data.TextLineDataset('secondary.csv') dataset = tf.data.Dataset.zip((so...

7得票2回答
使用PIP仅安装适用于CPU的Tensorflow 2.x

如何只使用pip安装CPU版本的Tensorflow 2.x? 过去,可以安装这2个不同版本。 由于我在没有Nvidia卡、只有不支持cuda的Intel卡的非GPU设备上运行脚本,所以我遇到了以下错误: 2020-04-14 23:28:14.632879:W tensorflow/str...

7得票1回答
在同样的顺序下对两个张量进行洗牌

如上所述。我已经尝试了那些方法,但都无果。 tf.random.shuffle( (a,b) ) tf.random.shuffle( zip(a,b) ) 我过去会把它们连接起来,然后进行洗牌,再取消连接/解包。但现在我遇到的情况是(a)是4D秩张量,而(b)是1D,所以无法连接。 ...