MLgym,一个用于分布式机器学习模型训练的python框架。

mlgym的Python项目详细描述


MLgym公司

基于python框架的分布式机器学习模型训练研究。在

CircleCI

MLgym的核心功能是在多个gpu上以比例分割的方式运行Pytorch模型的网格搜索,并使用DashifyML集中存储结果。在

此外,MLgym还提供以下主要功能:

  • 结果的再现性,这是由于完整的实验规范,包括数据集、预处理例程、模型体系结构、损失函数、度量等等。在
  • 组件注册表以注册具有依赖项的自定义组件。例如,可以定义一个新的预处理例程组件。该组件可能依赖于一个迭代器组件,如实验配置中所指定的。在运行时,这些组件是动态实例化的。在

请注意,目前此代码应视为实验性代码,尚未准备好投入生产。

安装

安装MLgym有两种选择,最简单的方法是从pip存储库安装:

pip install mlgym

对于最新版本,可以通过cd直接从源代码安装到根文件夹,然后运行

^{pr2}$

使用

注意:这个框架仍在大量开发中,主要用于研究项目。它很可能没有bug,接口仍然可以更改。

有关用法,请参阅此example。在

版权所有

版权所有(c)2020 Max Lübbering,Rajkumar Ramamurthy

有关许可证,请参见:https://github.com/le1nux/mlgym/blob/master/LICENSE

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
在OSGI中使用cxf生成的客户端时出现Java类装入器问题和JaxB异常   java为什么要在javamail中迭代多部分电子邮件中的部分?   并发编程问题   JFileChooser&&System中未调用java windowClosing。退出功能不正常?   SQL查询的java语法分析   java如何使用AspectJ声明字段上的警告   什么是java向量。元素()C#等价物   java解析Android应用程序中tornado web服务中的CSV文件   java我试过c2dm,我需要服务器端   java调整JPanel大小以适应新的JLabel图标   Java与Python脚本的通信   java使用Saxon通过XSLT生成URL   java net::ERR_complete_CHUNKED_编码200(OK)来自struts应用程序中的tomcat   java如何为我的窗格设置不同的位置?   java使用Cypher Neo4j获取给定类型的所有节点(从SQL世界中的tablename中选择*)   nio使用Java解析文件值   java使用WSDL生成REST客户端会是错误的方向吗?   java如何在我的应用程序中构建类映射?   java按钮。setEnabled在第一个循环中不起作用   xPath适用于最后一页,但不适用于第一页