测试时间增加抛出值错误

Question

我正在尝试在我的分类器上使用测试时间增强：

log_preds,y = learn.TTA(scale=1.1, ds_type=DatasetType.Valid, with_loss=True)

这是它抛出的错误：

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-14-f33d9737819a> in <module>()
----> 1 log_preds,y = learn.TTA(scale=1.1, ds_type=DatasetType.Valid, with_loss=True)
      2 probs = np.mean(np.exp(log_preds),0)
      3 
      4 accuracy(probs, y)

ValueError: too many values to unpack (expected 2)

最初，我试图找到一种在 TTA 中使用我的自定义测试集的方法，但不知道该怎么做，DatasetType.Test 抛出了一个错误，所以我决定使用 DatasetType.Valid 在运行 8 个 epoch 之后，我得到了上面的错误。

Answer 1

错误消息表明 learn.TTA return 超过 2 个值，但您只得到了 log_preds 和 y。你想知道 learn.TTA return

到底是做什么的

测试时间增加抛出值错误

Test Time Augmentation throwing a value error

python

computer-vision

deep-learning

pytorch

fast-ai