将密集网与 fastai 一起使用答案

【问题标题】：Using densenet with fastai将密集网与 fastai 一起使用
【发布时间】：2021-10-13 17:56:46
【问题描述】：

我正在尝试使用 fast.ai 库训练一个密集网络模型。我检查了文档，并设法使其适用于 resnet50。但是，对于densenet，似乎找不到模块。

如forum 所述，我尝试使用arch=models.dn121。但我得到了同样的错误。

有人可以帮忙吗？

代码如下：

learn = create_cnn(data, base_arch=models.densenet201, metrics=accuracy, model_dir="/tmp/model/")

这是错误：

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-23-cb9ab3a79572> in <module>()
----> 1 learn = create_cnn(data, base_arch=models.densenet201, metrics=accuracy, model_dir="/tmp/model/")

AttributeError: module 'fastai.vision.models' has no attribute 'densenet201'

【问题讨论】：

标签： python deep-learning conv-neural-network fast-ai densenet

【解决方案1】：

根据fast.ai论坛上的这个post，这是将densenet与fast.ai一起使用的解决方案：

from torchvision.models import densenet121

def dn121(pre): return children(densenet121(pre))[0]

learn = create_cnn(data, dn121)

【讨论】：