使用tfrecord构建数据集

背景 最近在打比赛,听说tfrecord的格式可以以二进制的方式高效存储数据,并利用protobuf协议读取数据,所以就想着把训练代码改成tf

获取模型的中间层输出

背景 最近在疯狂搭模型,突然有人来问我要我模型的中间层输出的embedding,于是,我就研究了一下怎么获取模型的中间层输出。 代码实现 1 2 3 4