属性错误：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 Traceback（最近一次调用最后一次）

属性错误：设备

我该如何解决这个问题？

Answer 1

试试这个：

1. 先将数据转换成databunch

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

2. 取决于硬件。 （我正在使用 cpu 进行处理）。

   进口火炬 data.device = torch.device('cpu')

3. 呼叫学习者。

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