0

我可以使用例如squeezenet 进行训练，但我是新手，所以我无法弄清楚其他架构（如darknet、xception 或xresnet）需要成功实例化。

在我的笔记本中，我尝试过：

arch = models.Darknet(num_classes = 2, num_blocks = 4) // whats numblocks
learn2 = cnn_learner(data = data, base_arch = arch , metrics=[accuracy, error_rate], pretrained = True)

我得到错误：

----> 1 arch = models.Darknet(num_classes = 2,num_blocks = 4)
      2 learn2 = cnn_learner(data = data, base_arch = arch , metrics=[accuracy, error_rate], pretrained = True)

/usr/local/lib/python3.6/dist-packages/fastai/vision/models/darknet.py in __init__(self, num_blocks, num_classes, nf)
     31         super().__init__()
     32         layers = [conv_bn_lrelu(3, nf, ks=3, stride=1)]
---> 33         for i,nb in enumerate(num_blocks):
     34             layers += self.make_group_layer(nf, nb, stride=2-(i==1))
     35             nf *= 2

TypeError: 'int' object is not iterable```
4

1 回答 1

0

从文档

创建一个暗网，其块的大小在 num_blocks 中给出，以 num_classes 结尾并使用 nf 初始特征。Darknet53 使用 num_blocks = [1,2,8,8,4]。

在此处了解有关暗网的更多信息

于 2019-05-26T22:52:24.487 回答