- 博客(3)
- 资源 (1)
- 问答 (2)
- 收藏
- 关注
RSS订阅原创 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
1072
原创 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
969
1
c++可不可以用map实现递归的数据结构
2021-01-21
TA创建的收藏夹 TA关注的收藏夹
TA关注的人