训练速度很慢

[Ghustwb]

训练数据6万张图片，batch size设置512，训练PNet，读取数据就花了4个小时。迭代速度 20step 耗时4分钟。这个在哪里可以改吗

[libohit]

我用wider训练没有这么慢呢！

[Ghustwb]

@libohit 我用了自己的数据，没有用wider。 训练数据的数量大概6万张图片，在加载数据过程中很慢，然后生成样本保存在本地的时候直接把硬盘存满了。正样本加上负样本就已经有五百万了。这个代码里面加载数据的部分没有使用pytorch的dataloader的api，而是自己实现的，不知道作者为什么不使用api而是选择自己去实现数据输入。

[kuaikuaikim]

@Ghustwb 加载数据可以选用队列(pytorch内置队列功能), 硬件方面推荐使用闪存吧，文件系统推荐ext4或者F2FS。

[renyanfei12321]

想问为什么没有使用torch自带的dataloader，自己写的列表读的太慢了，而且占用内存太高