用户可以将自有数据转换成MindRecord数据集,然后使用MindDataset类进行加载并进一步进行预处理操作。
1. 生成MindRecord示例
2. 多进程并发生成MindRecord示例
参考代码见附件
3. 生成MindRecord过程中可能遇到的问题
- 数据集太大,FileWriter.commit()的时候爆内存该怎么解决
- 答:可以组织一批数据之后,就调用
.write_raw_data(data_list)
接口写入,再组织一批数据,再调用.write_raw_data(data_list2)
,...,最后再调用.commit()
完成。即:组织数据 -> write_raw_data(...) -> 组织数据 -> write_raw_data(...) -> ... -> commit(),具体参照 Step1 中示例。 参考链接:华为云论坛_云计算论坛_开发者论坛_技术论坛-华为云