fastai 系列教程(二)- 快速入门 MNIST 示例

PyTorch入门实战教程

我们在上文中介绍了 fastai 的安装,本文将带领大家通过 MNIST 的例子快速上手 fastai。

引入包

首先我们引入必要的一些包:

fastai 中最重要的是 Leaner 类,我们先来熟悉一下 Learner 中的参数:

结果如下:

感兴趣的读者可以在这里查看官方文档。

创建一个 DataBunch

图像数据可以是一个标记好的文件夹,也可以是一个单独的文件和一个 csv 文件。

输出结果:

创建一个 DataBunch:

输出结果:

开始训练

创建一个 Learner 类,调用 fit 函数进行拟合:

输出结果:

 

PyTorch入门实战教程
除特别注明外,本站所有文章均为 PyTorch 中文网原创,转载请注明出处:https://www.pytorchtutorial.com/fastai-tutorial-2-overview-mnist/

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部