用户可以将自有数据转换成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(),具体参考链接:华为云论坛_云计算论坛_开发者论坛_技术论坛-华为云 中示例。
-
附件下载
附件请前往 https://www.hiascend.com/forum/thread-0230106819649046082-1-1.html03 下载。