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类型,错误消失。
具体原因还未搞清楚。先记录一下。