属性错误: fast.ai学习者中的设备

Question

我正在使用resnet18构建一个cnn学习者，代码如下：

from fastai.vision import *

data = (ImageList.from_df(train_df, path='/working/')
   .split_by_rand_pct(0.2)
   .label_from_df(label_delim=','))

learner = cnn_learner(data, models.resnet18, metrics=[accuracy])

我得到以下错误：

AttributeError跟踪(最近一次调用)

AttributeError:设备

我怎么解决这个问题？

Answer 1

尝试如下：

1. 首先将数据转换为数据库

数据=( path='/working/').split_by_rand_pct(0.2).label_from_df(label_delim='，').databunch()))，ImageList.from_df(Train_df))

1. 视硬件而定。(我正在使用cpu进行处理)。

进口火炬data.device =data.device

1. 给学习者打电话。

学习者= metrics=accuracy) (数据、models.resnet18、models.resnet18)