1. 首页
  2. 数据库
  3. 其它
  4. keras使用Sequence类调用大规模数据集进行训练的实现

keras使用Sequence类调用大规模数据集进行训练的实现

上传者: 2020-12-22 23:28:38上传 PDF文件 55.62KB 热度 11次
使用Keras如果要使用大规模数据集对网络进行训练,就没办法先加载进内存再从内存直接传到显存了,除了使用Sequence类以外,还可以使用迭代器去生成数据,但迭代器无法在fit_generation里开启多进程,会影响数据的读取和预处理效率,在本文中就不在叙述了,有需要的可以另外去百度。 下面是我所使用的代码 class SequenceData(Sequence): def __init__(self, path, batch_size=32): self.path = path self.batch_size = batch_size f = open(pat
用户评论