- 博客(3)
- 资源 (1)
- 问答 (2)
- 收藏
- 关注
原创 keras和Tensorflow的模型拆解或减少输出
有时候,我们需要减少预训练模型的输出个数。其实只用一行代码就可以实现。没必要对模型先进行拆分,再进行组装。比如我们先创建一个二输出的模型from tensorflow.keras import *import tensorflow as tffrom tensorflow.keras.layers import *INPUT=Input((32,32,3))x=Conv2D(4,2,p...
2020-04-10 07:31:35 1071
原创 tensorflow2.1的一次由numpy数据集引起的内存溢出
x与y都是numpy数据,x,y都是由1千张图片组成采用下面这条代码创建数据集。train_dataset=tf.data.Dataset.from_tensor_slices((x,y))在训练的过程中程序内存占用一直在增长,直到程序因为内存溢出而崩溃。官方教程的解释是上述这种由numpy创建数据集的方法只适用于小型数据,因为这种数据集被植入TensorFlow graph中,且数据会被...
2020-04-02 01:38:24 966 1
c++可不可以用map实现递归的数据结构
2021-01-21
TA创建的收藏夹 TA关注的收藏夹
TA关注的人