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

PyTorch入门实战教程

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

引入包

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

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

结果如下:

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

创建一个 DataBunch

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

输出结果:

创建一个 DataBunch:

输出结果:

开始训练

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

输出结果:

 

PyTorch入门实战教程

发表评论

电子邮件地址不会被公开。 必填项已用*标注

返回顶部