超简单!用pytorch训练自己的数据集

在pytorch官网上的tutoral中,教程给了一个例子,训练一个分类器,下载CIFAR10数据集,用一个简单的CNNC网络训练模型。在我们下载的CIFAR10数据集中,已标签的数据被打包并封装在data文件夹里。 我们可以在这个例子的基础上,把自己的数据放到其模型下,并实现一个简易的分类器。 首先是配置pytorch的各种环境,有很多教程的,他们讲的又详细又好,我就不多讲啦。 先导入各种模块
相关文章
相关标签/搜索