pytorchtypeerror:batchmustcontai

2019-11-19  本文已影响0人  FrankXu0808

错误信息:

 TypeError: batch must contain tensors, numbers, dicts or lists;  found <object>

在将numpy矩阵进行dataloader数据载入后

train_data = DataLoader(fortrain, batch_size=128, shuffle=True)

通过迭代器取值时,报上述错误。

而如果fortrain是普通的python list类型,错误消失。

具体原因还未搞清楚。先记录一下。

上一篇 下一篇

猜你喜欢

热点阅读