草庐IT

TensorFlow2

全部标签

python - TensorFlow 分配内存 : Allocation of 38535168 exceeds 10% of system memory

使用ResNet50预训练的权重我正在尝试构建一个分类器。代码库完全在Keras高级TensorflowAPI中实现。完整代码发布在下面的GitHub链接中。源代码:ClassificationUsingRestNet50Architecture预训练模型的文件大小为94.7mb。我加载了预训练的文件new_model=Sequential()new_model.add(ResNet50(include_top=False,pooling='avg',weights=resnet_weight_paths))并拟合模型train_generator=data_generator.flo

python - Windows 上的 TensorFlow 版本 1.0.0-rc2 : "OpKernel (' op: "BestSplits" device_type: "CPU"') for unknown op: BestSplits"with test code

我在Windows7SP1x64Ultimate(Python3.5.2|Anacondacustom(64-bit))上安装了TensorFlow版本1.0.0-rc2,使用:pipinstall--upgradehttps://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.0rc2-cp35-cp35m-win_amd64.whl当我尝试从https://web.archive.org/web/20170214034751/https://www.tensorflow.org/get_started/os_

python - 如何使用 TensorFlow GPU?

如何在Python3.6x64中使用TensorFlowGPU版本而不是CPU版本?importtensorflowastfPython正在使用我的CPU进行计算。我可以注意到它,因为我有一个错误:YourCPUsupportsinstructionsthatthisTensorFlowbinarywasnotcompiledtouse:AVX2我已经安装了tensorflow和tensorflow-gpu。如何切换到GPU版本? 最佳答案 按照本教程TensorflowGPU我做到了,效果很好。注意!-安装9.0版!Tensorf

python - Tensorflow : What is the relationship between . ckpt 文件和 .ckpt.meta 和 .ckpt.index 和 .pb 文件

我使用saver=tf.train.Saver()来保存我训练的模型,我得到了三种文件,命名为:.ckpt.meta.ckpt.index.ckpt.data还有一个文件叫:检查点与.ckpt文件有什么关系?我看到有人保存的模型只有.ckpt文件,我不知道如何制作。如何将模型保存为.pb文件? 最佳答案 .ckpt文件是saver.save(sess)的旧版本输出,相当于您的.ckpt-data(见下文)“检查点”文件只是在这里告诉一些TF函数哪个是最新的检查点文件。.ckpt-meta包含元图,即计算图的结构,没有变量的值(基本上

python - Tensorflow 中 tf.contrib 模块的用途是什么?

我很好奇tf.contrib是什么,以及为什么代码会包含在TensorFlow中,而不是包含在主存储库中。此外,查看示例here(来自tensorflowmaster分支),我想找到tf.contrib.layers.sparse_column_with_hash_bucket的来源。这似乎是一些很酷的例程,但我想确保它们正确使用队列等,用于预取/预处理示例,以便在生产环境中实际使用它们。似乎记录在案here,但它来自tflearn项目,但tf.contrib.layers.sparse_column_with_hash_bucket似乎也不在该存储库中。

python - Tensorflow 中的正确批量归一化功能是什么?

在tensorflow1.4中,我发现了两个执行批量标准化的函数,它们看起来一样:tf.layers.batch_normalization(link)tf.contrib.layers.batch_norm(link)我应该使用哪个功能?哪个更稳定? 最佳答案 只是添加到列表中,还有更多方法可以在tensorflow中进行批处理规范:tf.nn.batch_normalization是一个低级操作。调用者自己负责处理mean和variance张量。tf.nn.fused_batch_norm是另一个低级操作,与前一个类似。不同之处

python - 将 tensorflow 模型存储在内存中

我正在编写的程序涉及在运行时在模型之间切换。我目前正在使用Saver从磁盘中保存/加载模型,如下所示:https://www.tensorflow.org/api_docs/python/state_ops/saving_and_restoring_variables#Saver.模型相当小,可以存储在内存中,所以我想知道是否有人知道在内存中存储和恢复这些模型而不是将它们保存到磁盘的方法。我尝试修改tensorflow源以将模型保存到内存中,但是gen_io_ops似乎是在编译期间生成的。另一种可能的方法是使用内存映射文件。有谁知道更简单的方法吗? 最佳答案

python - 为什么我的 Deep Q Network 没有掌握一个简单的 Gridworld (Tensorflow)? (如何评估 Deep-Q-Net)

我尝试熟悉Q-learning和深度神经网络,目前尝试实现PlayingAtariwithDeepReinforcementLearning.为了测试我的实现并尝试使用它,我坚持尝试了一个简单的网格世界。我有一个NxN网格,从左上角开始,在右下角结束。可能的Action有:左、上、右、下。尽管我的实现与this非常相似(希望它是一个好的)它似乎没有学到任何东西。看看它需要完成的总步数(我猜网格大小为10x10的平均值约为500,但也有非常低和高的值),它比其他任何东西都更加随机。我在使用和不使用卷积层的情况下尝试了它,并使用了所有参数,但老实说,我不知道我的实现是否有问题或者它需要训练

python - 在TensorFlow中,变量名称中的 ":0"是什么意思?

importtensorflowastfwithtf.device('/gpu:0'):foo=tf.Variable(1,name='foo')assertfoo.name=="foo:0"withtf.device('/gpu:1'):bar=tf.Variable(1,name='bar')assertbar.name=="bar:0"上面的代码返回true。我这里用withtf.device来说明“:0”并不代表变量位于特定的设备上。那么“”是什么意思:0"在变量名中(本例中为foo和bar)? 最佳答案 这与底层API中张

python - 带有 bool 张量的 tensorflow 索引

在numpy中,有两个相同形状的数组x和y,可以像这样y[x>1]。您如何在tensorflow中获得相同的结果?y[tf.greater(x,1)]不起作用,tf.slice也不支持这样的东西。现在有没有办法使用bool张量进行索引,还是目前不受支持? 最佳答案 试试:ones=tf.ones_like(x)#createatensorallonesmask=tf.greater(x,ones)#booleantensor,mask[i]=Trueiffx[i]>1slice_y_greater_than_one=tf.boole