码迷,mamicode.com
首页 > 其他好文 > 详细

torchline:让Pytorch使用的更加顺滑

时间:2019-12-13 23:18:56      阅读:102      评论:0      收藏:0      [点我收藏+]

标签:ali   创建   ase   sea   hub   哈哈   mars   基于   它的   

torchline地址:https://github.com/marsggbo/torchline

相信大家平时在使用Pytorch搭建网络时,多少还是会觉得繁琐,因为我们需要搭建数据读取,模型,训练,checkpoints保存等等一系列模块。每当切换到新的任务后很多情况下之前的代码不能复用,或者说要复用就需要做很多地方的修改,到最后还不如重新写一遍。所幸,pytorch_lightning让这一过程简化了很多,相信如果你用过这个库你也会体验到它的方便性。但是torchline的存在是让你使用Pytorch更加的顺滑舒畅。

torchline基于pytorch_lightning (PL)开发,整个库的结构设计借鉴了detectron2,具体可以阅读下面几篇文章进行了解:

如下图所示,灰色部分 PL 可以自动完成。我们需要做的,差不多也就加载数据、定义模型、确定训练和验证过程
技术图片

torchline则进一步简化,而且可以让你的模型复用性更高。

以构建模型为例进行大致的介绍(细节可以去github查看),假如你之前创建了一个MyModel的模型,之后如果你想使用这个模型,你只需要在config文件中将MODEL.NAME修改成MyModel(即只是修改字符串的值,之后torchline会自动切换模型)。

总的来说,pytorch_lightning有的torchline肯定都有哈哈哈,但是使用起来代码复用性和易用性更高,欢迎去github品尝,觉得好用麻烦star,也欢迎issue讨论。



MARSGGBO?原创


如有意合作,欢迎私戳

邮箱:marsggbo@foxmail.com


2019-12-13 21:31:15



torchline:让Pytorch使用的更加顺滑

标签:ali   创建   ase   sea   hub   哈哈   mars   基于   它的   

原文地址:https://www.cnblogs.com/marsggbo/p/12037480.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!