![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch-lightning
文章平均质量分 92
Tina姐
微信公众号【医学图像人工智能实战营】,医学图像处理领域,坚持已实践为主,手把手带你做项目,打比赛,写论文。
展开
-
pytorch lightning 手写数字分类实例 (三)
今天通过手写数字来学习如何利用pytorch-lightning进行分类代码同第二部分的差不多,新增了断点训练和测试部分。项目使用jupyter notebook演示此部分代码很简单,小白也能上手,赶快来试一试吧~~~文章目录SET SEEDdataloadertrainingtesting测试结果显示import torchfrom torch import nnimport pytorch_lightning as plfrom torch.utils.data import DataL原创 2021-05-29 16:34:11 · 1464 阅读 · 5 评论 -
pytorch-lightning入门(一)—— 初了解
最近PyTorch Lightning的风很大,来看看为啥它这么火????文章目录1 什么是pytorch-lighting2 如何将PyTorch代码组织到Lightning中2.1 安装 PyTorch Lightning2.1 定义LightningModule1 什么是pytorch-lightingpytorch-lighting(简称pl),它其实就是一个轻量级的PyTorch库,用于高性能人工智能研究的轻量级PyTorch包装器。缩放你的模型,而不是样板。它可以清晰地抽象和自动化ML原创 2021-05-19 16:08:12 · 38881 阅读 · 4 评论 -
如何从Pytorch 到 Pytorch Lightning (二) | 简要介绍
这篇文章主要介绍为什么使用pytorch时,需要使用Lightning的最常见问题。PyTorch非常易于使用,可以构建复杂的AI模型。但是一旦研究变得复杂,并且将诸如多GPU训练,16位精度和TPU训练之类的东西混在一起,用户很可能引入Bug。PyTorch Lightning完全解决了这个问题。Lightning会构建您的PyTorch代码,以便可以抽象出训练的细节。这使得AI研究可扩展且可快速迭代。文章目录1 谁适用PyTorch Lightning?1 谁适用PyTorch Lightnin翻译 2021-05-21 10:31:17 · 2001 阅读 · 1 评论