keras fit_generator 并行
效果比较
- 样本量:1000张图片
- 模型: MobileNetV2
- epochs: 5
- CPU: 4核,3.4GHz
- GPU: None
可能数据量过小,并行的效果不是太明显。
数据读取方式 | workers | use_multiprocessing | 耗时/s |
---|---|---|---|
内存读取 | 0 | True | 1797 |
keras.utils.Sequence | 0 | False | 1475 |
keras.utils.Sequence | 4 | True |
参考: