美文网首页
pytorchtypeerror:batchmustcontai

pytorchtypeerror:batchmustcontai

作者: FrankXu0808 | 来源:发表于2019-11-19 15:21 被阅读0次

    错误信息:

     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类型,错误消失。

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

    相关文章

      网友评论

          本文标题:pytorchtypeerror:batchmustcontai

          本文链接:https://www.haomeiwen.com/subject/dyowictx.html