做 data preprocessing 和跑批量任务的时候经常会遇到下面的情况:
为解决上述烦恼,做了一个并行框架,项目地址: https://github.com/JeffersonQin/akasaka
代码非常简单但重在实用,已发布 PyPi 。
例子: https://github.com/JeffersonQin/akasaka/blob/master/examples/print.py
还支持 PyTorch CUDA 的 spawning, 在多卡集群上,用户可以只提供 load_model
, get_dataloader
, execute()
函数,即可多卡自动数据分割 + 并发执行,一步到位解决重复劳动
这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。
V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。
V2EX is a community of developers, designers and creative people.