Open yifa-n opened 7 years ago
背景:我将大约50W张图片做一个vgg的finetune。 使用,fit_generator,这个图片读取速度感觉太慢了。 硬件是K80的GPU,nvidia-smi看GPU的使用率很低,接近0. 是不是很多时间浪费在了数据的IO上面?
caffe之类的框架有lmdb等快速索引的工具。 keras文章中支持h5文件。能不能给出一个使用h5的demo ~~~
背景:我将大约50W张图片做一个vgg的finetune。 使用,fit_generator,这个图片读取速度感觉太慢了。 硬件是K80的GPU,nvidia-smi看GPU的使用率很低,接近0. 是不是很多时间浪费在了数据的IO上面?
caffe之类的框架有lmdb等快速索引的工具。 keras文章中支持h5文件。能不能给出一个使用h5的demo ~~~