![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow学习笔记
Life will be better
这个作者很懒,什么都没留下…
展开
-
解决tensorflow的tfrecords文件在训练时所占内存过大的问题
tfrecords文件在训练时所占内存过大的原因项目中需要把tiff格式的图像先制作成tensorflow的tfrecords格式,然后在训练时读取tfrecords文件为tf.data.Dataset 进行训练。而在读取tfrecords文件生成Dataset时一般都需要设置shuffle的大小,目的是在训练过程中,每个step都随机从Dataset中随机的获取一个batch的数据进行训练,能...原创 2019-11-03 16:04:13 · 2760 阅读 · 2 评论 -
tensorflow中keras的GlobalAveragePooling2D和AveragePooling2D的区别
平均池化CNN中是常用的操作,下面介绍一下tensorflow中keras的GlobalAveragePooling2D和AveragePooling2D的返回的tensor的维度的区别。GlobalAveragePooling2D是平均池化的一个特例,它不需要指定pool_size和strides等参数,操作的实质是将输入特征图的每一个通道求平均得到一个数值。它的输入和输出维度为: Inp...原创 2019-11-03 15:27:17 · 11521 阅读 · 0 评论 -
tensorflow中使用keras进行多gpu训练并进行模型保存
本文主要讲解了在tensorflow中使用keras进行多gpu训练的一些注意事项,分为三个部分使用多gpu训练的好处keras如何使用多gpu训练并使用检查点保存模型使用model.save保存模型后加载继续训练或者预测的注意事项1. 使用多gpu训练的好处(1)多gpu并行计算加速训练。单机多gpu训练实际上是将batch_size平均划分到多个gpu上进行并行计算,比如batc...原创 2019-11-03 12:35:23 · 2016 阅读 · 0 评论 -
keras tensorflow 多gpu训练
【转载来源】https://blog.csdn.net/u012862372/article/details/80367607 问题描...转载 2020-03-22 17:01:59 · 421 阅读 · 0 评论